poj 3321 Apple Tree(线段树)

题目大意：

        有一棵树，有n个节点，开始时每个分叉点有1个苹果，现在有m个操作，Q x代表查询x节点及x节点的子节点的苹果总和，C x代表将节点x的状态翻转，有苹果变成没苹果，没苹果变成有苹果。

解题思路：

        从节点1开始，进行一次深搜，根据遍历时间为每个节点编号，进入一个新节点时时间加1，离开时，时间标为从上一个节点离开的时间。那么每个节点的进入时间和离开时间就组成了一个区间[l,r]，且当前节点的有无苹果状态存在第l个点上，这个节点及子节点的苹果总和就是[l,r]上的苹果总和。所以问题就变成了一个线段树问题。

注意点：

        本题会卡vector，不建议使用。

代码：

#include <iostream>
#include <stdio.h>
#include <vector>

using namespace std;

struct range{
int l,r;
};

struct node{
int l,r;
int sum;
};

struct tNode{
int to,next;
tNode(){to = -1;next = -1;}
};

int n,m,nowPos = 0;;
tNode son[100005];
range a[100005];
node tree[400005];

void dfs(int x){
a[x].l = ++nowPos;//进入新节点时时间加1
if(son[x].to != -1){
int now = son[x].to;
while(now != -1){
dfs(now);
now = son[now].next;
}
}
a[x].r = nowPos;//离开时时间为从上一个节点离开的时间
}

void buildTree(int root,int l,int r){
tree[root].l = l;
tree[root].r = r;
tree[root].sum = r - l + 1;
if(l != r){
int mid = (l+r)>>1;
buildTree(2*root,l,mid);
buildTree(2*root+1,mid+1,r);
}
}

void updata(int root,int pos){
if(tree[root].l == pos && tree[root].r == pos){
tree[root].sum = 1 - tree[root].sum;
return;
}
int mid = (tree[root].l + tree[root].r)>>1;
if(pos <= mid){
updata(2*root,pos);
}
else{
updata(2*root+1,pos);
}
tree[root].sum = tree[2*root].sum + tree[2*root+1].sum;
}

int query(int root,int l,int r){
if(tree[root].l == l && tree[root].r == r){
return tree[root].sum;
}
int mid = (tree[root].l + tree[root].r)>>1;
if(r <= mid){
return query(2*root,l,r);
}
else if(l > mid){
return query(2*root+1,l,r);
}
else{
return query(2*root,l,mid) + query(2*root+1,mid+1,r);
}
}

int main()
{
char s[5];
int x,y;
scanf("%d",&n);
for(int i = 0; i < n-1; i ++){
scanf("%d%d",&x,&y);
son[y].next = son[x].to;
son[x].to = y;
}
dfs(1);
buildTree(1,1,n);
scanf("%d",&m);
for(int i = 0; i < m; i ++){
scanf("%s %d",s,&x);
if(s[0] == 'C'){
updata(1,a[x].l);
}
else{
printf("%d\n",query(1,a[x].l,a[x].r));
}
}
return 0;
}