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 655000int 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]));}