CSUOJ 1869 树上最大数

Description

现在给你一棵根节点编号为1的树，每个节点上都有对应的权值， 求出树上每一层的最大值。（根节点所在位置视为第一层，由此可推，根节点的儿子是处于第二层，etc）

Input

多组数据读入 第一行输入一个正整数n表示树上有n个节点（n<=100000)，下一行输入n个正整数v1,v2....vn表示n个节点上对应的权值（vi<=100000)， 再下一行输入n-1个正整数（第i个数字v表示，节点i+1的父亲是v，保证v<i+1）

Output

从树自顶向下输出每一层的最大值是多少，每个值之间用空格隔开，全部输出后末尾换行。

Sample Input

6
10 6 8 7 3 2
1 2 1 4 4

Sample Output

10 7 8

Hint

Source

中南大学第十一届大学生程序设计竞赛网络预选赛

Author

Forget_ever

思路：扫描所有节点，确定当前节点的层数，并更新该层数上的最大值。当然也有用ＤＦＳ或者ＢＦＳ的做法的，但我觉得并不需要用到这个，毕竟每次给的上一层的结点都是之前已经确定了层数的（题目中：保证v<i+1）

代码：

#include<cstdio>
#include<cstring>
#include<set>
#include<algorithm>
#include<string>
using namespace std;
typedef long long LL;
int tree[1000010],bef[1000010],kind[1000010],maxn[1000010];
//tree构造数，实际上并不需要扫描该树
//bef记录上一层的结点
//kind记录层数
//maxn记录每一层的最大值 
int main()
{
int n;
while(scanf("%d",&n)!=EOF){
memset(kind,0,sizeof(kind));
memset(maxn,0,sizeof(maxn));
for(int i=0;i<n;i++) scanf("%d",&tree[i]);
for(int i=1;i<n;i++) scanf("%d",&bef[i]);
for(int i=1;i<n;i++){
kind[i]=kind[bef[i]-1]+1;
if(maxn[kind[i]]<tree[i])
maxn[kind[i]]=tree[i];
}
printf("%d",tree[0]);
for(int i=1;kind[i]>0&&maxn[i]>0;i++){
printf(" %d",maxn[i]);
}
printf("\n");
}
return 0;
}