bzoj 1103: [POI2007]大都市meg【dfs序+树状数组】

很明显的暗示,就是在树的dfs序上维护树状数组,加减的时候差分即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=500005;
int n,m,h[N],cnt,s[N],top,tot,t[N],fa[N],l[N],r[N];
struct qwe
{
    int to,ne;
}e[N];
int read()
{
	int r=0,f=1;
	char p=getchar();
	while(p>'9'||p<'0')
	{
		if(p=='-')
			f=-1;
		p=getchar();
	}
	while(p>='0'&&p<='9')
	{
		r=r*10+p-48;
		p=getchar();
	}
	return r*f;
}
void add(int u,int v)
{
	cnt++;
	e[cnt].ne=h[u];
	e[cnt].to=v;
	h[u]=cnt;
}
void update(int x,int val)
{
    for(int i=x;i<=n+n;i+=(i&(-i)))
		t[i]+=val;
}
int ques(int x)
{
    int r=-1;
    for(int i=x;i;i-=(i&(-i)))
		r+=t[i];
	return r;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
		int x=read(),y=read();
		add(x,y),add(y,x);
    }
    s[++top]=1;
    while(top)
    {
        int u=s[top],f=fa[top--];
        if(!l[u])
        {
            l[u]=++tot;
            s[++top]=u;
            for(int i=h[u];i;i=e[i].ne)
				if(e[i].to!=f)
				{
					s[++top]=e[i].to;
					fa[top]=u;
				}
        }
        else
			r[u]=++tot;
    }
    for(int i=1;i<=n;i++)
    {
        update(l[i],1);
        update(r[i],-1);
    }
    scanf("%d",&m);
    for(int i=1;i<=n+m-1;i++)
    {
        char ch[5];
        scanf("%s",ch);
        if(ch[0]=='A')
        {
            int x=read(),y=read();
            update(l[y],-1);
			update(r[y],1);
        }
        else
        {
            int x=read();
			printf("%d
",ques(l[x]));
        }
    }
    return 0;
}
原文地址:https://www.cnblogs.com/lokiii/p/9641917.html