猴猴最喜欢在树上玩耍,一天猴猴又跳上了一棵树,这棵树有N个苹果,每个苹果有一个编号,分别为0~N-1,它们之间由N-1个树枝相连,猴猴可以从树枝的一端爬到树枝的另一端,所以猴猴可以从任意一个苹果的位置出发爬到任意猴猴想去的苹果的位置。猴猴开始在编号为K的苹果的位置,并且把这个苹果吃了,之后每一天猴猴都要去吃一个苹果,但是树上那么多苹果吃哪个呢?猴猴想到自己去吃苹果时一定会把路上遇到的苹果都吃掉,于是猴猴决定去吃能让自己这天吃的苹果数量最多的那个苹果,如果有多个苹果满足条件,猴猴就会去吃这些中编号最小的苹果,那么猴猴会按照什么顺序吃苹果呢?
输入
第一行两个数N和K。(N<=50000)
第2-N行,第i+1行的数字Ai表示i和Ai之间有一根树枝相连。
输出
每行一个数字,依次表示猴猴所在的位置
数据范围
对于30%的数据:N<=100
对于60%的数据:N<=1000
对于100%的数据:N<=50000,0<=K<N
输入样例
输入样例1
7 2
0
1
2
2
1
4
输入样例2
8 1
0
0
1
3
0
3
1
输入样例3
4 2
0
0
1
输出样例
输出样例1
2
0
6
3
5
输出样例2
1
2
4
5
6
7
输出样例3
2
3
样例解释
第一天最多可以吃到两个苹果,可以去0或6,去0
第二天最多吃两个,去6
第三天最多吃一个,去3
第四天最多吃一个,去5
//written by zzy
题目大意:
有一个树,每个点开始有个苹果,猴子会从一个点开始找一条有最多苹果的路,然后吃掉那一条路上的所有的苹果,求猴子吃那条路上最后苹果的顺序。
题解 :
首先不难想到猴子每次必到叶子结点,
将起始点看成根,
可以看成猴子每吃完一次后重回到根节点(因为路径上的苹果吃完了后没有贡献),
再遍历一次,求出每个点的深度和父亲,按深度把叶子结点从大到小排序,
然后可以倒过来想,让每个叶子结点往上跳,遇到跳过的点就停,记录跳的次数(即为从根吃到它的贡献值),
再按此值排序即可。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 50005
using namespace std;
int l,i,j,n,m,k,num,v;
int cnt[N],fa[N],d[N];
bool b[N],vis[N];
int next[2*N],list[2*N],y[2*N];
struct leafs {
int a,x,s;
}leaf[N];
void add(int x)
{
next[++l]=list[x];
list[x]=l;
}
bool cmp_x(leafs p,leafs q) {
return (p.x>q.x)||(p.x==q.x&&p.a<q.a);
}
bool cmp_s(leafs p,leafs q) {
return (p.s>q.s)||(p.s==q.s&&p.a<q.a);
}
void dfs(int father,int x)
{
for (int t=list[x],to=y[t];t;t=next[t],to=y[t])
if (b[to]) {
b[to]=false; fa[to]=x; d[to]=d[x]+1;
if (cnt[x]!=1) dfs(x,to);
}
}
int main()
{
scanf("%d%d",&n,&k);
printf("%d\n",k);
for (i=1;i<=n-1;i++) {
scanf("%d",&v);
y[i*2-1]=v; add(i);
y[i*2]=i; add(v);
cnt[i]++; cnt[v]++;
}
memset(b,true,sizeof(b));
b[k]=false; d[k]=0; cnt[k]=0;
dfs(-1,k);
for (i=0;i<=n-1;i++)
if (cnt[i]==1)
leaf[++num].a=i,leaf[num].x=d[i];
sort(leaf+1,leaf+num+1,cmp_x);
vis[k]=true;
for (i=1;i<=num;i++) {
int now=leaf[i].a;
while (!vis[now])
vis[now]=true,leaf[i].s++,now=fa[now];
}
sort(leaf+1,leaf+num+1,cmp_s);
for (i=1;i<=num;i++)
printf("%d\n",leaf[i].a);
}