【bzoj1103】【POI2007】【大都市】【meg】【人工栈】

题目大意

一棵有根树，每次修改一条边或询问跟到一个点的路径上有多少条边没被修改。

解题思路

可以用括号序表示树上的一条路径，访问时间戳+1，离开时间戳-1，修改一条边就是将深度大的点的时间戳都改为0，表示对后面没影响。最恶心的是要打人工栈。

code

#include
#include
#include
#include
#include
#define LL long long
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=250000,maxa=1000000000;
int n,m,gra,time,top,to[maxn+10],next[maxn+10],begin[maxn+10],dfn[maxn+10],lea[maxn+10],st[maxn+10],t[maxn*2+10];
void insert(int u,int v){
    to[++gra]=v;
    next[gra]=begin[u];
    begin[u]=gra;
}
void dfs(int s){
    st[++top]=s;
    for(;top;){
        if(begin[st[top]])st[top+1]=to[begin[st[top]]],begin[st[top]]=next[begin[st[top]]],top++,dfn[st[top]]=++time;
        else lea[st[top]]=++time,top--;
    }
}
void change(int p,int v){
    for(;p<=time;){
        t[p]+=v;
        p+=p&(-p);
    }
}
int qury(int p){
    int ans=0;
    for(;p>=1;){
        ans+=t[p];
        p-=p&(-p);
    }
    return ans;
}
int main(){
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    scanf("%d",&n);
    fo(i,1,n-1){
        int u,v;scanf("%d%d",&u,&v);
        if(u>v)swap(u,v);
        insert(u,v);
    }
    dfs(1);
    fo(i,2,n){
        change(dfn[i],1);
        change(lea[i],-1);
    }
    scanf("%d\n",&m);
    fo(i,1,n+m-1){
        char ch=getchar();int a,b;
        if(ch=='W'){
            scanf("%d\n",&a);
            printf("%d\n",qury(dfn[a]));
        }
        else{
            scanf("%d%d\n",&a,&b);
            change(dfn[b],-1);
            change(lea[b],1);
        }
    }
    return 0;
}