BZOJ 3631 链剖+差分

思路：

1.树链剖分+用带标记的线段树维护操作（复杂度O(nlog2n)）

2.树链剖分LCA（TarjanLCA等各种LCA）+差分 复杂度（O(n)—>O(nlogn)之间）

下面就说说怎么差分……

把mark[x]++,mark[y]++,mark[lca(xx,yy)]–,mark[fa[lca(xx,yy)]]–;

最后输出的时候要把 不是a[1]的节点–。

搞定~~~

//By SiriusRen
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 655000
int a[N],next[N],first[N],v[N],w[N],tot;
int size[N],deep[N],son[N],fa[N],top[N],p[N],cnt;
int n,xx,yy,mark[N];
void add(int x,int y){v[tot]=y,next[tot]=first[x],first[x]=tot++;}
void dfs(int x){
    size[x]=1;
    for(int i=first[x];~i;i=next[i])
        if(v[i]!=fa[x]){
            deep[v[i]]=deep[x]+1,fa[v[i]]=x;
            dfs(v[i]);
            size[x]+=size[v[i]];
            if(size[son[x]]<size[v[i]])son[x]=v[i];
        }
}
void dfs2(int x,int tp){
    top[x]=tp,p[++cnt]=x;
    if(son[x])dfs2(son[x],tp);
    for(int i=first[x];~i;i=next[i])
        if(v[i]!=fa[x]&&v[i]!=son[x])
            dfs2(v[i],v[i]);
}
void dfs3(int x){
    for(int i=first[x];~i;i=next[i])
        if(v[i]!=fa[x])
            dfs3(v[i]),mark[x]+=mark[v[i]];
}
int lca(int x,int y){
    int fx=top[x],fy=top[y];
    while(fx!=fy){
        if(deep[fx]<deep[fy])swap(x,y),swap(fx,fy);
        x=fa[fx],fx=top[x];
    }
    if(deep[x]>deep[y])swap(x,y);
    return x;
}
int main(){
    memset(first,-1,sizeof(first));
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<n;i++){
        scanf("%d%d",&xx,&yy);
        add(xx,yy),add(yy,xx);
    }
    dfs(1),dfs2(1,1);
    for(int i=2;i<=n;i++){
        mark[a[i]]++,mark[a[i-1]]++;
        int t=lca(a[i],a[i-1]);mark[t]--,mark[fa[t]]--;
    }
    dfs3(1);
    for(int i=1;i<=n;i++)printf("%d\n",mark[i]-(i!=a[1]));
}