POJ3321

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;
}