E. Little Elephant and Tree dfs序+线段树

题目链接： http://codeforces.com/contest/258/problem/E

题意：

你现在有一棵树，每个结点初始情况下都是一个空集，现在有 $q$ 次操作，第 $i$ 次操作会往以 $a,b$ 为根的子树上的每个结点中放入数字 $i$ ，最后问你对于每个结点 $x(x\in [1,n])$ 有多少个结点和它有至少一个相同数字。

做法：

要用dfs序的线段树进行统计，这个还是比较明显的，但是该如何做每一个结点呢，自然是按照 $dfs$ 的顺序来，在进入结点的时候将对应的点进行更新，出去的时候清除。

代码

#include <bits/stdc++.h>
#define lson rt<<1
#define rson rt<<1|1
#define mid (l+r)/2
#define rep(i,a,b) for(int i = (int)a;i<=(int)b;i++)
using namespace std;
typedef long long ll;
const int maxn=100005;
const int maxm=200005;
const int inf=1e9+7;
vector<int> hav[maxn];
int st[maxn],en[maxn],tot,cnt,ans[maxn];
int n,m,head[maxn],to[maxm],nex[maxm];
int sum[maxn<<3],laz[maxn<<3];
void push_up(int l,int r,int rt){
    if(laz[rt]) sum[rt]=r-l+1;
    else {
        sum[rt]=sum[lson]+sum[rson];
    }
}
void update(int l,int r,int rt,int ql,int qr,int v){
    if(ql<=l&&r<=qr){
        laz[rt]+=v;
        push_up(l,r,rt);
        return ;
    }
    if(ql<=mid) update(l,mid,lson,ql,qr,v);
    if(qr>mid) update(mid+1,r,rson,ql,qr,v);
    push_up(l,r,rt);
}
void add(int u,int v){
    to[cnt]=v;nex[cnt]=head[u];
    head[u]=cnt++;
}
void init(int u,int f){
    st[u]=++tot;
    for(int i=head[u];~i;i=nex[i]){
        int v=to[i];
        if(v==f) continue;
        init(v,u);
    }
    en[u]=tot;
}
void dfs(int u,int f){
    for(auto x: hav[u]){
        update(1,tot,1,st[x],en[x],1);
    }
    ans[u]=max(sum[1]-1,0);
    for(int i=head[u];~i;i=nex[i]){
            int v=to[i];
        if(f==v) continue;
        dfs(v,u);
    }
    for(auto x: hav[u]){
        update(1,tot,1,st[x],en[x],-1);
    }
}
int main(){
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    rep(i,2,n){
        int x,y; scanf("%d%d",&x,&y);
        add(x,y);  add(y,x);
    }
    rep(i,1,m){
        int a,b; scanf("%d%d",&a,&b);
        hav[a].push_back(a); hav[b].push_back(a);
        hav[a].push_back(b); hav[b].push_back(b);
    }
    init(1,-1);
    dfs(1,-1);
    rep(i,1,n){
        printf("%d%c",ans[i],i==n?'\n':' ');
    }

    return 0;
}