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\n",ques(l[x]));
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/lokiii/p/9641917.html