题目
二叉树上有 n个节点,按从0到 n - 1编号,其中节点i的两个子节点分别是leftChild[i]和rightChild[i]。
只有 所有 节点能够形成且 只 形成 一颗有效的二叉树时,返回true;否则返回 false。
如果节点i没有左子节点,那么leftChild[i]就等于-1。右子节点也符合该规则。
注意:节点没有值,本问题中仅仅使用节点编号。
示例 1:
输入:n = 4, leftChild = [1,-1,3,-1], rightChild = [2,-1,-1,-1]
输出:true
示例 2:
输入:n = 4, leftChild = [1,-1,3,-1], rightChild = [2,3,-1,-1]
输出:false
示例 3:
输入:n = 2, leftChild = [1,0], rightChild = [-1,-1]
输出:false
示例 4:
输入:n = 6, leftChild = [1,-1,-1,4,-1,-1], rightChild = [2,-1,-1,5,-1,-1]
输出:false
static const int N = 1e4 + 5;
// 抓住一点: 一个合法的树, 边数等于结点数 - 1, 且无环无孤立的点
int inn[N], out[N];
bool validateBinaryTreeNodes(int n, vector<int>& leftChild, vector<int>& rightChild) {
int edges = 0;
for (int i = 0; i <= n - 1; ++i) {
if (leftChild[i] != -1) edges++, inn[leftChild[i]]++, out[i]++;
if (rightChild[i] != -1) edges++, inn[rightChild[i]]++, out[i]++;
}
// 还要排除环, 看是否存在孤立的点
if (edges != n - 1) return false;
// 特判
if (n == 1) return true;
for (int i = 0; i <= n - 1; ++i) {
if (inn[i] == 0 && out[i] == 0) return false;//排除双头
}
return true;
}
下面这种做法有问题
#include<vector>
using namespace std;
class MyClass
{
public:
bool validateBinaryTreeNodes(int n, vector<int>& leftChild, vector<int>& rightChild) {
vector<int> inDegree(n, 0);
bool findRoot = false;
int sum = 0;
for (int i = 0; i < n; ++i){
if (leftChild[i] != -1){
++inDegree[leftChild[i]];
}
if (rightChild[i] != -1){
++inDegree[rightChild[i]];
}
}
for (int i = 0; i < n; ++i){
if (!findRoot&&inDegree[i] == 0){
findRoot = true;
continue;
}
else if (findRoot && inDegree[i] == 0){
return false;
}
else if (inDegree[i] != 1){
return false;
}
sum+=inDegree[i];
}
if (n - 1 != sum){
return false;
}
return true;
}
};
一些学习心得和体会。不敢妄称原创,基本是学习笔记。致谢相关大神的引用