递增二叉树-网易游戏
题目描述
给定一个二叉树, 每个节点有一个正整数权值。若一棵二叉树, 每一层的节点权值和都严格小于下一层的结点权值和, 责成这棵二叉树为递增二叉树。
现在给你一棵二叉树,你需要判断其是不是一棵递增二叉树。
输入描述
输入的第一行是一个正整数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;
}