题目
二叉树上有 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
提示:
1 <= n <= 10^4
leftChild.length == rightChild.length == n
-1 <= leftChild[i], rightChild[i] <= n - 1
思路
看题目和例子总结
- 二叉树
- 一个节点存在一个以上的父节点 return false
- 闭环 return false
- 一个以上的联通量, return false
使用并查集
可以解决上面2-4点情况。
- 第二种情况
在union
的时候判断一下节点是否已经关联了父节点就行了 - 第三种情况。
find
一下父子节点的根节点,判断是否相等就可以知道父子节点是否存在闭环 - 第四种情况。
统计一下执行union
的次数,就可以知道只剩下多少个联通量。
代码
class Solution {
int ans;
public boolean validateBinaryTreeNodes(int n, int[] leftChild, int[] rightChild) {
int[] fa = new int[n];
ans = n;
for (int i = 0; i < n; ++i) fa[i] = i;
for (int i = 0; i < n; ++i) {
if (union(fa, i, leftChild[i]) || union(fa, i, rightChild[i])) return false;
}
if (ans > 1) return false;
return true;
}
public int find(int[] fa, int n) {
return fa[n] == n ? n : find(fa, fa[n]);
}
public boolean union(int[] fa, int f, int c) {
if (c == -1) return false;
if (fa[c] != c) return true;
int a = find(fa, f), b = find(fa, c);
if (a == b) return true;
fa[b] = a;
ans--;
return false;
}
}