[JLOI2014]松鼠的新家

题目大意：
　　给你一棵n个结点的带点权的树，指定访问结点的顺序。
　　每次访问走最短路径，并将经过结点的点权+1。
　　问最后各个结点的点权。

思路：
　　树链剖分。
　　每次访问求一下LCA即可，同时维护一下线段树。
　　因为最后只需要统计一次，中间没有询问，因此可以最后在处理一下lazy标记。

#include<cstdio>
#include<cctype>
#include<vector>
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
const int N=300001;
int a[N];
std::vector<int> e[N];
inline void add_edge(const int &u,const int &v) {
    e[u].push_back(v);
    e[v].push_back(u);
}
int n,dep[N],par[N],size[N],son[N],top[N],id[N],node[N],ans[N];
void dfs1(const int &x) {
    size[x]=1;
    for(unsigned i=0;i<e[x].size();i++) {
        const int &y=e[x][i];
        if(y==par[x]) continue;
        par[y]=x;
        dep[y]=dep[x]+1;
        dfs1(y);
        size[x]+=size[y];
        if(size[y]>size[son[x]]) {
            son[x]=y;
        }
    }
}
void dfs2(const int &x) {
    id[x]=++id[0];
    node[id[x]]=x;
    if(x==son[par[x]]) {
        top[x]=top[par[x]];
    } else {
        top[x]=x;
    }
    if(son[x]) dfs2(son[x]);
    for(unsigned i=0;i<e[x].size();i++) {
        const int &y=e[x][i];
        if(y==par[x]||y==son[x]) continue;
        dfs2(y);
    }
}
class SegmentTree {
    #define _left <<1
    #define _right <<1|1
    private:
        int val[N<<2];
    public:
        void modify(const int &p,const int &b,const int &e,const int &l,const int &r,const int &x) {
            if(b==l&&e==r) {
                val[p]+=x;
                return;
            }
            const int mid=(b+e)>>1;
            if(l<=mid) modify(p _left,b,mid,l,std::min(mid,r),x);
            if(r>mid) modify(p _right,mid+1,e,std::max(mid+1,l),r,x);
        }
        void stat(const int &p,const int &b,const int &e) {
            if(b==e) {
                ans[node[b]]=val[p];
                return;
            }
            val[p _left]+=val[p];
            val[p _right]+=val[p];
            const int mid=(b+e)>>1;
            stat(p _left,b,mid);
            stat(p _right,mid+1,e);
        }
    #undef _left
    #undef _right
};
SegmentTree t;
inline void modify(int u,int v) {
    t.modify(1,1,n,id[v],id[v],-1);
    while(top[u]!=top[v]) {
        if(dep[top[u]]<dep[top[v]]) std::swap(u,v);
        t.modify(1,1,n,id[top[u]],id[u],1);
        u=par[top[u]];
    }
    if(dep[u]<dep[v]) std::swap(u,v);
    t.modify(1,1,n,id[v],id[u],1);
}
int main() {
    n=getint();
    for(register int i=1;i<=n;i++) a[i]=getint();
    for(register int i=1;i<n;i++) {
        add_edge(getint(),getint());
    }
    dfs1(1);
    dfs2(1);
    for(register int i=1;i<n;i++) {
        modify(a[i],a[i+1]);
    }
    t.stat(1,1,n);
    for(register int i=1;i<=n;i++) {
        printf("%d
",ans[i]);
    }
    return 0;
}