Apple Tree （dfs序+线段树）

Apple Tree （dfs序+线段树）

/*dfs+线段树（单点修改+区间查询）*/
#include "iostream"
#include "vector"
#include "map"
#include "cstdio"
#include "cstring"

using namespace std;
typedef long long ll;
const int maxn = 100005;
int n,m,_time,cnt;
int num[maxn],in[maxn],out[maxn],last[maxn];
int sum[maxn<<2];
struct data{
    int to,_next;
}e[maxn<<1];

void add(int u,int v){
    e[++cnt].to = v; e[cnt]._next=last[u]; last[u]=cnt;
    e[++cnt].to = u; e[cnt]._next=last[v]; last[v]=cnt;
}

void dfs_arr(int x,int fa){
    in[x]=++_time;
    num[_time]=x;
    for(int i=last[x];i;i=e[i]._next){
        if( e[i].to!=fa ) dfs_arr(e[i].to,x);
    }
    out[x]=_time;
}

void build(int pt,int l,int r){
    int mid=(l+r)>>1;
    if( l==r ){
        sum[pt]=1;
        return ;
    }
    build(pt<<1,l,mid);
    build(pt<<1|1,mid+1,r);
    sum[pt] = sum[pt<<1]+sum[pt<<1|1];
}

int query(int pt,int L,int R, int l,int r){
    if( L<=l&&R>=r ) return sum[pt];
    int mid=(l+r)>>1;

    int sum=0;
    if( L<=mid ) sum+=query(pt<<1,L,R,l,mid);
    if( R>mid ) sum+=query(pt<<1|1,L,R,mid+1,r);
    return sum;
}

void modify(int pt,int x,int l,int r)
{
    int mid=(l+r)>>1;
    if( l==r )
    {
        sum[pt]^=1;
        return ;
    }
    if( x<=mid ) modify(pt<<1,x,l,mid);
    else modify(pt<<1|1,x,mid+1,r);
    sum[pt] = sum[pt<<1]+sum[pt<<1|1];
    return ;
}


int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++ ){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    dfs_arr(1,0);
    scanf("%d",&m);
    char ch[10];
    build(1,1,n);
    for(int i=1;i<=m;i++){
        scanf("%s",ch+1);
        int x;
        scanf("%d",&x);
        if( ch[1]=='Q' ) printf("%d
",query(1,in[x],out[x],1,n));
        else modify(1,in[x],1,n);
    }
    return 0;
}