关闭

【Codevs1228】苹果树(dfs序+树状数组)

87人阅读 评论(0) 收藏 举报
分类:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100010;
int n,m,a1,a2,in[maxn],out[maxn],head[maxn],ncnt,cnt,c[maxn],vis[maxn];
struct edge{
    int to,ne;
}e[maxn];

inline int lowbit(int x){return x&(x^(x-1));}

void add(int x,int y){
    e[++ncnt].to=y;
    e[ncnt].ne=head[x];
    head[x]=ncnt;
}

void dfs(int x){//深搜生成dfs序 
    in[x]=++cnt;//进的时间戳 
    for(int i=head[x];i;i=e[i].ne){
        dfs(e[i].to);
    }
    out[x]=cnt;//出的时间戳 
}

void change(int x,int k){
    while(x<=n){
        c[x]+=k;
        x+=lowbit(x);
    }
}

int query(int x){
    int ans=0;
    while(x){
        ans+=c[x];
        x-=lowbit(x); 
    }
    return ans;
}

void init(){
    for(int i=1;i<=n;i++){
        change(in[i],1);
        vis[i]=1;//每一个节点都有苹果 
    }
}

int main(){
    cin>>n;
    for(int i=1;i<n;i++){
        scanf("%d%d",&a1,&a2);
        add(a1,a2);
    }
    vis[n]=1;
    dfs(1);
    init();
    cin>>m;
    while(m--){
        char ch;
        int x;
        cin>>ch;
        cin>>x;
        if(ch=='C'){
            if(vis[x]){
                change(in[x],-1);
            }else{
                change(in[x],1);
            }
            vis[x]=!vis[x];
        }else{
            printf("%d\n",query(out[x])-query(in[x]-1));
        }
    }
    return(0);
}
1
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:1905次
    • 积分:116
    • 等级:
    • 排名:千里之外
    • 原创:9篇
    • 转载:5篇
    • 译文:0篇
    • 评论:1条
    文章分类
    文章存档
    最新评论