[JLOI2014]松鼠的新家

来源:互联网 发布:mac axure 没响应 编辑:程序博客网 时间:2024/03/29 13:09

Description

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

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

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

Input

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

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

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

Output

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

Sample Input

51 4 5 3 21 22 42 34 5

Sample Output

12121

HINT

30%的数据,n<=4000

80%的数据,n<=50000

100%的数据,2<= n <=300000


#include<stdio.h>int b[21],n,cnt,f[300001][21],a[300001],fa[300001],last[300001],deep[300001];struct Edge{int to,next;}e[600001];void swap(int &m,int &k){int o=m;m=k;k=o;}void addage(int u,int v){e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;}void dfs(int x){for(int i=1;i<=18;i++)if(deep[x]>=b[i])f[x][i]=f[f[x][i-1]][i-1];for(int i=last[x];i;i=e[i].next)if(e[i].to!=f[x][0]){deep[e[i].to]=deep[x]+1;f[e[i].to][0]=x;dfs(e[i].to);}}int lca(int x,int y){if(deep[x]<deep[y])swap(x,y);int t=deep[x]-deep[y];for(int i=0;i<=18;i++)if(b[i]&t)x=f[x][i];for(int i=18;i>=0;i--)if(f[x][i]!=f[y][i]){x=f[x][i];y=f[y][i];}if(x==y)return x;return f[x][0];}void dfs2(int x){for(int i=last[x];i;i=e[i].next)if(e[i].to!=f[x][0]){dfs2(e[i].to);fa[x]+=fa[e[i].to];}}int main(){scanf("%d",&n);b[0]=1;for(int i=1;i<=20;i++)b[i]=b[i-1]<<1;for(int i=1;i<=n;i++)scanf("%d",&a[i]);int u,v;for(int i=1;i<n;i++){scanf("%d%d",&u,&v);addage(u,v);addage(v,u);}dfs(a[1]);    for(int i=1;i<n;i++){int t=lca(a[i],a[i+1]);fa[a[i]]++;fa[a[i+1]]++;fa[t]--;fa[f[t][0]]--;}dfs2(a[1]);for(int i=2;i<=n;i++)fa[a[i]]--;for(int i=1;i<=n;i++)printf("%d\n",fa[i]);}


原创粉丝点击