[POI2007]MEG-Megalopolis (树状数组,Dfs序)

Posted 2020-12-04 kv-stalin

题目描述

Solution

这道题考试的时候竟然没有仔细想,结果只拿了暴力分...
其实就是一个 DFS序+树状数组。

我们先把用 DFS 把它变成一个序列,同时记录它们的 (siz)。
那么我们每一次连一条边之后就是对它的子树产生影响。
在树状数组里面维护就好了。

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=250008;
struct sj{
    int to;
    int next;
}a[maxn*2];
int head[maxn],size;
int n,m;

void add1(int x,int y)
{
    a[++size].to=y;
    a[size].next=head[x];
    head[x]=size;
}

int id[maxn],tot;
int siz[maxn],ans[maxn];
void dfs(int x)
{
    id[x]=++tot;
    siz[x]=1;
    for(int i=head[x];i;i=a[i].next)
    {   
        int tt=a[i].to;
        if(!siz[tt])
        {
            ans[tt]=ans[x]+1;
            dfs(tt);
            siz[x]+=siz[tt];    
        }
    }
}

int c[maxn];
int lowbit(int x)
{return x&(-x);}
void add(int x,int z)
{for(int i=x;i<=n;i+=lowbit(i))c[i]+=z;}
int check(int x)
{
    int sum=0;
    for(int i=x;i>=1;i-=lowbit(i))
      sum+=c[i];
    return sum;
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {int x,y; scanf("%d%d",&x,&y);add1(x,y); add1(y,x);}
    dfs(1);
    for(int i=1;i<=n;i++)
    add(id[i],0);
    scanf("%d",&m);
    for(int i=1;i<=m+n-1;i++)
    {
        int x,y;char ch; cin>>ch;
        if(ch=='A')
        {
            scanf("%d%d",&x,&y);
            if(id[x]<id[y])swap(y,x);
            add(id[x],1);
            add(id[x]+siz[x],-1);
        }
        if(ch=='W')
        scanf("%d",&x),
            cout<<ans[x]-check(id[x])<<endl;
    }
}