poj3321 树状数组+dfs（时间戳）

最新推荐文章于 2020-08-18 10:33:44 发布

hill_555

最新推荐文章于 2020-08-18 10:33:44 发布

阅读量773

点赞数

CC 4.0 BY-SA版权

分类专栏：树状数组

本文链接：https://blog.youkuaiyun.com/hillcoder/article/details/8801904

树状数组专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种使用树状数组解决POJ3321问题的方法，该问题涉及在一棵树上进行苹果的增删操作，并查询特定子树内苹果的数量。通过深度优先搜索（DFS）为每个节点分配时间戳，利用树状数组快速计算区间和。

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

poj3321
树状数组

题意：给定一棵树，某些节点上有苹果，多次询问各子树上的节点数，并且在询问的中途随时可能新增和删除苹果。

分析：dfs遍历树的同时，对每个点标注时间，每个点有一个开始时间和一个结束时间，把这两个时间当做下标，该点的苹果个数（1或0）填入数组的两个对应位。子树中结点的时间段一定是根节点时间段的子段，所以求子树苹果数，只需求数组某区间的和即可。用树状数组比较快。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100005

int n;//n个苹果 树状数组边界
int m;//m个询问
//*******************************************
int tot;//邻接表边数量
int head[N];//边邻接表头
struct node
{
    int to,next;
}e[2*N];
void add(int u,int v)
{
    e[tot].to=v;
    e[tot].next=head[u];
    head[u]=tot++;
}
//*******************************************
int ti;//时间戳
bool vis[N]; //dfs用
int st[N],en[N];//dfs 时间
void dfs(int u)
{
    ti++;
    st[u]=ti;
    vis[u]=1;
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        if(!vis[e[i].to])
            dfs(e[i].to);
    }
    en[u]=ti;
}
//*******************************************
int cal[N];//树状数组用
int lowbit(int x) {return x&(-x);}
int getsum(int x)
{
    int s=0;
    for(;x>0;x-=lowbit(x)) s+=cal[x];
    return s;
}
void update(int x,int value)
{
    for(;x<=n;x+=lowbit(x)) cal[x]+=value;
}
//*******************************************
bool apple[N];//苹果有没有 题用
int main()
{
    int i,u,v;
    char s[5];
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    memset(vis,0,sizeof(vis));
    memset(apple,1,sizeof(apple));
    tot=0;
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    ti=0;
    dfs(1);
    for(i=1;i<=n;i++) update(i,1);
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s%d",s,&u);
        if(s[0]=='C')
        {
            if(apple[u])
            {
                update(st[u],-1);
                apple[u]=0;
            }
            else
            {
                update(st[u],1);
                apple[u]=1;
            }
        }
        else printf("%d\n",getsum(en[u])-getsum(st[u]-1));
    }
    return 0;
}