您的位置:首页 > 其它

ZOJ 3516 Tree of Three(DFS)

2012-05-10 21:28 375 查看
题目链接: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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: