POJ-3321-Apple Tree

最新推荐文章于 2025-03-07 14:39:34 发布

原创最新推荐文章于 2025-03-07 14:39:34 发布 · 446 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树状数组

ACM 同时被 2 个专栏收录

716 篇文章

订阅专栏

POJ

378 篇文章

订阅专栏

本文介绍了一种针对苹果树的更新与查询算法实现。通过深度优先搜索（DFS）为每个节点分配时间戳，利用树状数组进行高效的区间更新与查询操作。适用于解决区间修改与查询的问题。

这个题要求对苹果树进行更新与查找工作，先用DFS找出时间戳，然后树状数组进行更新操作即可

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=1e6+100;
struct node
{
    int l;
    int r;
}t[maxn];
int n,m,time,c[maxn*3],a[maxn],head[maxn],pnt[maxn],nxt[maxn],e;
void addedge(int u,int v)
{
    pnt[e]=v;nxt[e]=head[u];head[u]=e++;
}
void DFS(int u)
{
    t[u].l=time;
    for(int i=head[u];i!=-1;i=nxt[i])
	DFS(pnt[i]);
    t[u].r=time++;
}
int lowbit(int x)
{
    return x&(-x);
}
void change(int x)
{
    if(a[x])
	for(int i=x;i<time;i+=lowbit(i))
	    c[i]++;
    else
	for(int i=x;i<time;i+=lowbit(i))
	    c[i]--;
}
int  sum(int x)
{
    int ans=0;
    for(int i=x;i>0;ans+=c[i],i-=lowbit(i));
    return ans;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
	memset(head,-1,sizeof(head));
	e=0;
	time=1;
	for(int i=0;i<n-1;i++)
	{
	    int u,v;
	    scanf("%d%d",&u,&v);
	    addedge(u,v);
	}
	DFS(1);
	scanf("%d",&m);
	for(int i=1;i<=n;i++)
	{
	    a[i]=1;
	    change(i);
	}
	while(m--)
	{
	    char op[3];
	    int ita;
	    scanf("%s%d",op,&ita);
	    if(op[0]=='Q')
		printf("%d\n",sum(t[ita].r)-sum(t[ita].l-1));
	    else
	    {
		a[t[ita].r]^=1;
		change(t[ita].r);
	    }
	}
    }
    return 0;
}