递增二叉树-网易游戏

递增二叉树-网易游戏

题目描述

给定一个二叉树， 每个节点有一个正整数权值。若一棵二叉树， 每一层的节点权值和都严格小于下一层的结点权值和， 责成这棵二叉树为递增二叉树。

现在给你一棵二叉树，你需要判断其是不是一棵递增二叉树。

输入描述

输入的第一行是一个正整数T（0<T<=50）。接下来有T组样例，对于每组样例，输入的第一行是一个整数N， 表示树的节点个数（0<N<=1000, 节点编号为0到N-1）。接下来是N行， 第k行表述编号为k的节点，输入的格式：Value, left, right, 其中value表示权值， 是一个不超过5000的自然数； left和right分别表示该节点的左右子编号， 如果不存在左子或右子， 则left或right为-1；

2
8
2 -1 -1
1 5 3
4 -1 6
2 -1 -1
3 0 2
2 4 7
7 -1 -1
2 -1 -1
8
21 6 -1
52 4 -1
80 0 3
31 7 -1
21 -1 -1
59 -1 -1
50 5  -1
48 -1 1

输出描述

对于每-组样例，输出一个字符串，是递增二叉树则输出YES， 反之NO

YES
NO

代码

#include <bits/stdc++.h>

using namespace std;

struct TreeNode{
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int x): val(x), left(nullptr), right(nullptr){
	}
};

int toNum(string &s){
	int num;
	stringstream ss;
	ss<<s;
	ss>>num;
	return num;
}

int main(){
	int T;
	cin >> T;
	while(T--){
		int n;
		cin >> n;
		// 读入数据  
		vector<TreeNode*> v(n); //存放每个节点 
		vector<int> lnode(n); // 存放左节点序号 
		vector<int> rnode(n); // 存放右节点序号 
		for(int i=0; i<n; i++){
			int value;			
			cin >> value >> lnode[i] >> rnode[i];
			TreeNode* node = new TreeNode(value); // 构建二叉树的节点， 默认左右子树为空； 
			v[i] = node; // 将其放入vector中 
		}
		int tmp;
		vector<bool> flag(n, true); // 如果flag[i]为true，则其为根节点 
		for(int i=0; i<n; i++){
			if(lnode[i]!=-1){
				tmp = lnode[i];
				v[i]->left = v[tmp]; // 指向其左子树 
				flag[tmp] = false;
			} 
			if(rnode[i]!=-1){
				tmp = rnode[i];
				v[i]->right = v[tmp]; // 指向其右子树 
				flag[tmp]=false;
			}		 				
		}
		
		// 找到根节点 
		int i=0;
		for(; i<n; i++){
			if(flag[i]==true){
				break;
			}
		}

		TreeNode* root = v[i];
		bool res = true;
		queue<TreeNode*> Q;
		vector<int> num;
		
		Q.push(root);
		while(!Q.empty()){
			int temp = 0;
			int count = Q.size();
			while(count--){
				TreeNode* pNode = Q.front();
				Q.pop();
				temp += pNode->val;
				if(pNode->left)
					Q.push(pNode->left);
				if(pNode->right)
					Q.push(pNode->right);
			}
			num.push_back(temp);		
		}
		for(int i=1; i<num.size(); i++){
			if(num[i]<=num[i-1]){
				res = false;
				break;
			}	
		}
		if(res)
			cout << "YES" << endl;
		else
			cout << "NO" << endl;
									
	}
	return 0;
}