题目链接:Click here~~
题意:给一棵树,有n个节点(编号:0~n-1),每个节点有一个权值,0为树根,找出某个节点所在子树的最大的3个值。
解题思路:树根已经确定,所以问题也变的比较简单,只需要先确定叶子节点的最值,依次向父节点传递影响,直到根节点。则很容易想到用深搜来实现。
#include <vector>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define Max 10005
#define CLR(arr,val) memset(arr,val,sizeof(arr))
using namespace std;
vector <int> V[Max];
int n,ans[Max][7];
bool cmp(int a,int b)
{
return a > b;
}
void dfs(int x)
{
while(!V[x].empty())
{
int p = V[x].back();
V[x].pop_back();
dfs(p);
for(int i=3;i<6;i++)
ans[x][i] = ans[p][i-3];
sort(ans[x],ans[x]+6,cmp);
}
}
int main()
{
int a,b,Q,x;
while(~scanf("%d",&n))
{
CLR(ans,-1);
scanf("%d",&ans[0][0]);
for(int i=1;i<n;i++)
{
scanf("%d%d",&a,&b);
V[a].push_back(i);
ans[i][0] = b;
}
dfs(0);
scanf("%d",&Q);
while(Q--)
{
scanf("%d",&x);
if(ans[x][2] == -1)
printf("-1\n");
else
printf("%d %d %d\n",ans[x][0],ans[x][1],ans[x][2]);
}
}
return 0;
}