bzoj3631 松鼠的新家

Description

松鼠的新家是一棵树，前几天刚刚装修了新家，新家有n个房间，并且有n-1根树枝连接，每个房间都可以相互到达，且俩个房间之间的路线都是唯一的。天哪，他居然真的住在“树”上。松鼠想邀请小熊维尼前来参观，并且还指定一份参观指南，他希望维尼能够按照他的指南顺序，先去a1，再去a2，……，最后到an，去参观新家。

可是这样会导致维尼重复走很多房间，懒惰的维尼不听地推辞。可是松鼠告诉他，每走到一个房间，他就可以从房间拿一块糖果吃。维尼是个馋家伙，立马就答应了。

现在松鼠希望知道为了保证维尼有糖果吃，他需要在每一个房间各放至少多少个糖果。因为松鼠参观指南上的最后一个房间an是餐厅，餐厅里他准备了丰盛的大餐，所以当维尼在参观的最后到达餐厅时就不需要再拿糖果吃了。

Input

第一行一个整数n，表示房间个数

第二行n个整数，依次描述a1-an

接下来n-1行，每行两个整数x，y，表示标号x和y的两个房间之间有树枝相连。

Output

一共n行，第i行输出标号为i的房间至少需要放多少个糖果，才能让维尼有糖果吃。

tarjan离线求出a_i和a_i+1的lca并在树上差分记录a_i和a_i+1间的路径，第二次dfs由差分求出答案。

#include<cstdio>
const int N=300005;
int n,a,b;
int nx[N*2],es[N*2],e0[N],ep=1;
int dep[N],fa[N],f[N],as[N],ar[N],s[N],ans[N];
bool d[N];
int read(){
    int x=0,c=getchar();
    while(c>'9'||c<'0')c=getchar();
    while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();
    return x;
}
int stk[16];
void print(int x){
    int p=0;
    if(!x)stk[p++]=0;
    while(x){
        stk[p++]=x%10;
        x/=10;
    }
    for(--p;~p;--p)putchar(stk[p]+'0');
    putchar(10);
}
int get(int x){
    int a=x,c;
    while(a!=f[a])a=f[a];
    while(a!=(c=f[x]))f[x]=a,x=c;
    return a;
}
void dfs(int w,int pa,int de){
    d[w]=1;
    fa[w]=pa;
    dep[w]=de++;
    if(d[w-1]){
        int c=get(w-1);
        s[w]++;s[w-1]++;
        s[c]--;s[fa[c]]--;
    }
    if(d[w+1]){
        int c=get(w+1);
        s[w]++;s[w+1]++;
        s[c]--;s[fa[c]]--;
    }
    for(int i=e0[w];i;i=nx[i]){
        int u=es[i];
        if(u==pa)continue;
        dfs(u,w,de);
        f[u]=w;
    }
}
int dfs2(int w){
    int v=s[w];
    for(int i=e0[w];i;i=nx[i]){
        int u=es[i];
        if(u==fa[w])continue;
        v+=dfs2(u);
    }
    ans[w]=v-(w>1&&w<=n);
    return v;
}
int main(){
    n=read();
    for(int i=1;i<=n;i++)f[i]=i;
    for(int i=1;i<=n;i++)as[i]=read();
    for(int i=1;i<=n;i++)ar[as[i]]=i;
    for(int i=1;i<n;i++){
        a=ar[read()];b=ar[read()];
        es[ep]=b;nx[ep]=e0[a];e0[a]=ep++;
        es[ep]=a;nx[ep]=e0[b];e0[b]=ep++;
    }
    dfs(1,0,0);
    dfs2(1);
    for(int i=1;i<=n;i++)print(ans[ar[i]]);
    return 0;
}