Poj 3321
题目意思 给你一颗树 每个节点都有苹果,有M个操作
C i 表示 如果 i 有苹果 那就拿走i的 否则就增加一个
Q i 查询 i 以及子树下的苹果数量
#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#define maxn 100005
using namespace std;
int n;
struct node
{
int l,r,sum;
}q[maxn<<2];
struct link
{
int to,next;
}edges[maxn];
int head[maxn],edge;
void init()
{
memset(head,-1,sizeof(head));
edge=0;
}
void add(int u,int v)
{
edges[edge].to=v;
edges[edge].next=head[u];
head[u]=edge++;
}
int S[maxn],E[maxn],id,m;
void build(int rt,int l,int r)
{
q[rt].l=l;
q[rt].r=r;
q[rt].sum=1;
if(l==r)
return;
build(rt<<1,l,(l+r)/2);
build(rt<<1|1,(l+r)/2+1,r);
q[rt].sum=q[rt<<1].sum+q[rt<<1|1].sum;
}
void updata(int rt,int i)
{
if(q[rt].r==q[rt].l)
{
q[rt].sum^=1;
return;
}
int mid=(q[rt].r+q[rt].l)/2;
if(mid>=i){
updata(rt<<1,i);
}else{
updata(rt<<1|1,i);
}
q[rt].sum=q[rt<<1].sum+q[rt<<1|1].sum;
}
int query(int rt,int l,int r)
{
int ll=q[rt].l,rr=q[rt].r,mid=(ll+rr)>>1;
if(l==ll&&r==rr)
return q[rt].sum;
if(r<=mid)
return query(rt<<1,l,r);
else if(l>mid)
return query(rt<<1|1,l,r);
else
return query(rt<<1,l,mid)+query(rt<<1|1,mid+1,r);
}
void dfs(int x)
{
S[x]=++id;
for(int i=head[x];i!=-1;i=edges[i].next)
{
int v=edges[i].to;
dfs(v);
}
E[x]=id;
}
int main()
{
scanf("%d",&n);
init();
for(int i=0;i<n-1;i++)
{
int u,v;
scanf("%d%d",&u,&v);
add(u,v);
}
scanf("%d",&m);
char ch[10];
int v;
dfs(1);
build(1,1,n);
for(int i=0;i<m;i++)
{
scanf("%s%d",ch,&v);
if(ch[0]=='Q'){
printf("%d\n",query(1,S[v],E[v]));
}
else{
updata(1,S[v]);
}
}
return 0;
}