ccz181078

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: :: 管理 ::

Description

松鼠的新家是一棵树,前几天刚刚装修了新家,新家有n个房间,并且有n-1根树枝连接,每个房间都可以相互到达,且俩个房间之间的路线都是唯一的。天哪,他居然真的住在“树”上。松鼠想邀请****前来参观,并且还指定一份参观指南,他希望**能够按照他的指南顺序,先去a1,再去a2,……,最后到an,去参观新家。
可是这样会导致**重复走很多房间,懒惰的**不听地推辞。可是松鼠告诉他,每走到一个房间,他就可以从房间拿一块糖果吃。**是个馋家伙,立马就答应了。
现在松鼠希望知道为了保证**有糖果吃,他需要在每一个房间各放至少多少个糖果。因为松鼠参观指南上的最后一个房间an是餐厅,餐厅里他准备了丰盛的大餐,所以当**在参观的最后到达餐厅时就不需要再拿糖果吃了。

Input

第一行一个整数n,表示房间个数
第二行n个整数,依次描述a1-an
接下来n-1行,每行两个整数x,y,表示标号x和y的两个房间之间有树枝相连。

Output

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

tarjan离线求出ai和ai+1的lca并在树上差分记录ai和ai+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;
}

 

posted on 2016-02-28 15:10  nul  阅读(165)  评论(0编辑  收藏  举报