bzoj1103[POI2007]大都市meg

链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1103

题意：中文题。

分析：题目要求的是每次都是从节点1计算到节点x的距离，我们可以利用长度为2*n的dfs序的性质，因为在dfs序中我们会保存节点i进入时间l[i]和出去时间r[i]，那么我们可以设置一个树状数组记录前缀和，我们将所有的l[i]位置+1，所有的r[i]位置-1。这样有什么用呢？我们会发现从根节点到节点i的路径上的和就是l[i]的前缀和，这是因为我们将其他的点都看成环用+1-1抵消掉了，在删边的时候将被删的那个偏大的节点的l[i]和r[i]还原为0即可。

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=250100;
const int MAX=151;
const int mod=100000000;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=998244353;
const ll INF=10000000010;
typedef double db;
typedef unsigned long long ull;
int tot,u[N],v[N],pre[N];
void add_b(int x,int y) {
    v[tot]=y;pre[tot]=u[x];u[x]=tot++;
}
char s[5];
int p,ti,l[N],r[N],f[2*N];
void dfs(int x) {
    l[x]=++ti;
    for (int i=u[x];i!=-1;i=pre[i]) dfs(v[i]);
    r[x]=++ti;
}
void add(int x,int y) {
    for (;x<=p;x+=x&(-x)) f[x]+=y;
}
int getsum(int x) {
    int ret=0;
    for (;x;x-=x&(-x)) ret+=f[x];
    return ret;
}
int main()
{
    int i,n,m,x,y;
    scanf("%d", &n);
    tot=0;memset(u,-1,sizeof(u));
    for (i=1;i<n;i++) {
        scanf("%d%d", &x, &y);
        if (x>y) swap(x,y);add_b(x,y);
    }
    ti=0;p=2*n;dfs(1);
    memset(f,0,sizeof(f));
    for (i=2;i<=n;i++) {
        add(l[i],1);add(r[i],-1);
    }
    scanf("%d", &m);m+=n-1;
    while (m--) {
        scanf("%s", s);
        if (s[0]=='W') {
            scanf("%d", &x);
            printf("%d\n", getsum(l[x]));
        } else {
            scanf("%d%d", &x, &y);
            if (x>y) swap(x,y);
            add(l[y],-1);add(r[y],1);
        }
    }
    return 0;
}