[POJ 3321]Apple Tree dfs序+树状数组

最新推荐文章于 2025-08-14 18:10:12 发布

玄学家CS

最新推荐文章于 2025-08-14 18:10:12 发布

阅读量353

点赞数 1

CC 4.0 BY-SA版权

分类专栏：树状数组文章标签： dfs序算法

本文链接：https://blog.youkuaiyun.com/qq_38329396/article/details/78219459

树状数组专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用树状数组结合深度优先搜索（DFS）的方法来高效处理树形结构上的节点权值翻转及子树权值求和问题。通过构建边的数据结构，利用DFS确定子树范围，并借助树状数组实现快速更新和查询。

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

给定一棵树，树上权值初始时为1。

给出两种操作：

1.将单点权值取反。

2.查询子树的权值和。

我们可以通过dfs将其映射到一个树状数组里，记录子树的起始位置以及终止位置。直接暴力改logn查询即可。

#include<stdio.h>
#include<vector>
using namespace std;
const int Maxn=2e5+5;
int n,m,Timer=0,tot=0;
int c[Maxn],in[Maxn],out[Maxn],f[Maxn];
bool b[Maxn];
struct edge{
    int nex,to;
}e[Maxn];
inline void add(int u,int v){
    e[++tot]=(edge){f[u],v};
    f[u]=tot;
}
void dfs(int x){
    in[x]=++Timer;
    for(int i=f[x];i;i=e[i].nex){
        int v=e[i].to;
        if(in[v]) continue;
        dfs(v);
    }
    out[x]=Timer;
}
inline int lowbit(int x){return x&-x;}
void update(int x,int v){ for(;x<=n;x+=lowbit(x))c[x]+=v; }
int query(int x){int sum=0;for(;x;x-=lowbit(x))sum+=c[x];return sum;}
int main(){
    scanf("%d",&n);
    int u,v;
    for(int i=1;i<n;++i){
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    dfs(1);
    scanf("%d",&m);
    for(int i=1;i<=n+1;++i) update(i,1),b[i]=1;
    char s[4];
    for(int i=1;i<=m;++i){
        scanf("%s%d",s,&u);
        if(s[0]=='C') update(in[u],b[u]?-1:1),b[u]=1-b[u];
        else printf("%d\n",query(out[u])-query(in[u]-1));
    }
    return 0;
}