题目描述
松鼠的新家是一棵树,前几天刚刚装修了新家,新家有n个房间,并且有n-1根树枝连接,每个房间都可以相互到达,且俩个房间之间的路线都是唯一的。天哪,他居然真的住在”树“上。
松鼠想邀请小熊维尼前来参观,并且还指定一份参观指南,他希望维尼能够按照他的指南顺序,先去a1,再去a2,…,最后到an,去参观新家。可是这样会导致维尼重复走很多房间,懒惰的维尼不停地推辞。可是松鼠告诉他,每走到一个房间,他就可以从房间拿一块糖果吃。
维尼是个馋家伙,立马就答应了。现在松鼠希望知道为了保证维尼有糖果吃,他需要在每一个房间各放至少多少个糖果。
因为松鼠参观指南上的最后一个房间an是餐厅,餐厅里他准备了丰盛的大餐,所以当维尼在参观的最后到达餐厅时就不需要再拿糖果吃了。
输入输出格式
输入格式:
第一行一个整数n,表示房间个数第二行n个整数,依次描述a1-an
接下来n-1行,每行两个整数x,y,表示标号x和y的两个房间之间有树枝相连。
输出格式:
一共n行,第i行输出标号为i的房间至少需要放多少个糖果,才能让维尼有糖果吃。
今天学树上差分,做了道裸题…
看完题后的我:这题用树剖不是能水么233333
看完数据范围后的我:???完了树剖要被卡
网上说这题用树剖可以水过…数据这么渣的吗?
这是个树上差分裸题,没什么好说的
然后我打了一年…
智障错误不要学我。
#include<cstdio>
#include<iostream>
using namespace std;
const int maxn=300000+500;
struct Edge
{
int f;
int to;
int next;
}edge[maxn*2];
int n;
int tot=1;
int head[maxn];
void add_e(int f,int t)
{
edge[tot].f=f;
edge[tot].to=t;
edge[tot].next=head[f];
head[f]=tot;
tot++;
}
int deep[maxn];
int fa[maxn][31];
int seq[maxn];
void dfs(int t)
{
for(int i=head[t];i;i=edge[i].next)
{
Edge e=edge[i];
if(!deep[e.to])
{
deep[e.to]=deep[t]+1;
fa[e.to][0]=t;
dfs(e.to);
}
}
}
int tmp[maxn];
int cnt;
void fafa()
{
cnt++;
printf("cnt:%d",cnt);
for(int i=1;i<=n;i++)
printf("tmp:%d\n",tmp[i]);
}
void clr()
{
for(int i=1;i<=29;i++)
{
for(int j=1;j<=n;j++)
fa[j][i]=fa[fa[j][i-1]][i-1];
}
}
int lcaa(int x,int y)
{
if(deep[x]<deep[y])
swap(x,y);
for(int i=29;i>=0;i--)
{
if(deep[fa[x][i]]>=deep[y])
x=fa[x][i];
}
if(x==y)
return x;
for(int i=29;i>=0;i--)
if(fa[x][i]!=fa[y][i])
x=fa[x][i],y=fa[y][i];
return fa[x][0];
}
int ans[maxn];
int dep[maxn];
void dfs1(int t)
{
for(int i=head[t];i;i=edge[i].next)
{
Edge e=edge[i];
if(!dep[e.to])
{
dep[e.to]++;
dfs1(e.to);
tmp[t]+=tmp[e.to];
// fafa();
}
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&seq[i]);
for(int i=1;i<n;i++)
{
int a,b;
scanf("%d%d",&a,&b);
add_e(a,b);
add_e(b,a);
}
deep[1]=1;
dfs(1);
clr();
for(int i=1;i<n;i++)
{
int aa=seq[i],bb=seq[i+1];
int lca=lcaa(aa,bb);
// fafa();
tmp[aa]++,tmp[bb]++;
tmp[lca]--,tmp[fa[lca][0]]--;
ans[bb]--;
}
dep[1]=1;
dfs1(1);
for(int i=1;i<=n;i++)
printf("%d\n",tmp[i]+ans[i]);
// for(int i=1;i<=n;i++)
// printf("%d\n",ans[i]);
return 0;
}