【题目】Problem 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

解答：

import java.util.Scanner;
/*
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
 */
public class Test{
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		while(scanner.hasNext()) {
			int n = scanner.nextInt();
			for(int i = 0; i < n; i++) {
				int l = scanner.nextInt();
				
				int[][] nums = new int[l][2];
				nums[0][1] = -1;
				for(int j = 0; j < l; j++) {
					nums[j][0] = scanner.nextInt();
				}
				for(int j = 0; j < l - 1; j++) {
					int x = scanner.nextInt() - 1, y = scanner.nextInt() - 1;
					nums[y][1] = x;
				}
				if(l == 1 || l == 0) {
					System.out.println("YES");
					continue;
				}
				for(int j = 1; j < l; j++) {
					if(nums[j][0] < nums[nums[j][1]][0]) {
						System.out.println("NO");
						break;
					}else if(j == l - 1) System.out.println("YES");
				}
			}
		}
	}
}