【Codevs1228】苹果树（dfs序+树状数组）

最新推荐文章于 2022-05-05 11:28:08 发布

原创最新推荐文章于 2022-05-05 11:28:08 发布 · 492 阅读

0 ·

CC 4.0 BY-SA版权

题解专栏收录该内容

11 篇文章

订阅专栏

本文深入探讨了如何利用dfs序和树状数组解决苹果树问题，详细解析了算法思路和实现过程，帮助读者理解这两种数据结构在处理树形问题中的高效应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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