北邮 网研2014 B题 最小堆——不需要建树的二叉树题目

题目:
Problem B. 最小堆
题目描述
给定一棵带权二叉树,请判断它是不是一个最小堆。
一棵二叉树是一个最小堆,当且仅当对于树上任意一个节点,它的权值都小于或等于以它为根的子树中的所有权值。
输入格式
输入数据第一行是一个整数T(1<=T<=100),表示测试数据的组数。
对于每组测试数据:
第一行是一个整数N(1<=N<=100),表示树的节点个数。
接下来一行包含N个正整数,第i个整数valuei(1<=valuei<=1000)表示编号i的点的权值。
接下来N-1行,每行两个整数u和v(1<=u,v<=N, u!=v),表示节点u是节点v的父节点。
测试数据保证给定的一定是一棵二叉树,并且节点1是树的根结点。
输出格式
对于每组测试数据,如果给定的树是一个最小堆则输出Yes,否则输出No。
输入样例
3
1
10
3
10 5 3
1 2
1 3
5
1 2 3 4 5
1 3
1 2
2 4
2 5
输出样例
Yes
No
Yes
题解:
这个题看似需要建树在遍历检查是否符合题目要求,但是由二叉树的性质知,若在输入每条树的分支时判定每个根节点的权值大于子孙权值,那么这个树就一定符合要求。

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int t,n,m,u,v;
	int value[1005];
	cin>>t;
	while(t--)
	{
		cin>>n;
		for(int i=1;i<=n;i++) cin>>value[i];
		if(n==1) cout<<"Yes"<<endl;
		else
		{
			bool flag=true;
		    n--;
		    while(n--)
		    {
			    cin>>u>>v;
			    if(value[u]>value[v]) {flag=false;}//核心判断
		    }
		    if(flag)cout<<"Yes"<<endl;
		    
		    else cout<<"No"<<endl;
        }
	}
	return 0;
 } 

猜你喜欢

转载自blog.csdn.net/qq_38993096/article/details/88674088