C++刷leetcode（1361. 验证二叉树）【并查集】【并查集rank优化】

糖逗

题目描述：

1. 二叉树上有 n 个节点，按从 0 到 n - 1 编号，其中节点 i 的两个子节点分别是 leftChild[i] 和 rightChild[i]。
   
2. 
   
3. 只有 所有 节点能够形成且 只 形成 一颗 有效的二叉树时，返回 true；否则返回 false。
   
4. 
   
5. 如果节点 i 没有左子节点，那么 leftChild[i] 就等于 -1。右子节点也符合该规则。
   
6. 
   
7. 注意：节点没有值，本问题中仅仅使用节点编号。
   
8. 
   
9.  
   
10. 
    
11. 示例 1：
    
12. 
    
13. 
    
14. 
    
15. 输入：n = 4, leftChild = [1,-1,3,-1], rightChild = [2,-1,-1,-1]
    
16. 输出：true
    
17. 示例 2：
    
18. 
    
19. 
    
20. 
    
21. 输入：n = 4, leftChild = [1,-1,3,-1], rightChild = [2,3,-1,-1]
    
22. 输出：false
    
23. 示例 3：
    
24. 
    
25. 
    
26. 
    
27. 
    
28. 
    
29. 输入：n = 2, leftChild = [1,0], rightChild = [-1,-1]
    
30. 输出：false
    
31. 示例 4：
    
32. 
    
33. 
    
34. 
    
35. 输入：n = 6, leftChild = [1,-1,-1,4,-1,-1], rightChild = [2,-1,-1,5,-1,-1]
    
36. 输出：false
    
37.  
    
38. 
    
39. 提示：
    
40. 
    
41. 1 <= n <= 10^4
    
42. leftChild.length == rightChild.length == n
    
43. -1 <= leftChild[i], rightChild[i] <= n - 1
    
44. 
    
45. 来源：力扣（LeetCode）
    
46. 链接：https://leetcode-cn.com/problems/validate-binary-tree-nodes
    
47. 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

复制代码

1. class Solution {
   
2. public:
   
3.     int find_root(int x, vector<int> & father){
   
4.         if(x != father[x]) return find_root(father[x], father);
   
5.         return x;
   
6.     }
   
7.     bool uniton(int x, int y, vector<int> & father, vector<int> & rank){
   
8.         int temp1 = find_root(x, father);
   
9.         int temp2 = find_root(y, father);
   
10.         if(temp1 == temp2)return true;
    
11.         if(rank[temp1] < rank[temp2]){
    
12.             father[temp1] = temp2;
    
13.         }else if(rank[temp1] >= rank[temp2]){
    
14.             father[temp2] = temp1;
    
15.             if(rank[temp1] == rank[temp2]){
    
16.                 rank[temp2]++;
    
17.             }
    
18.         }
    
19.         return false;
    
20.     }
    
21.     bool validateBinaryTreeNodes(int n, vector<int>& leftChild, vector<int>& rightChild) {
    
22.         vector<int> father(n, 0);
    
23.         vector<int> rank(n, 0);
    
24.         for(int i = 0; i < n; i++){
    
25.             father[i] = i;
    
26.             rank[i] = 1;
    
27.         }
    
28.         for(int i = 0; i < n; i++){
    
29.             if(leftChild[i] != -1){
    
30.                 if(uniton(i, leftChild[i], father, rank)) return false;
    
31.             }
    
32.             if(rightChild[i] != -1){
    
33.                 if(uniton(i, rightChild[i], father, rank)) return false;
    
34.             }
    
35.         }
    
36.         set<int> group;
    
37.         for(int i = 0; i < n; i++){
    
38.             group.insert(find_root(i, father));
    
39.             if(group.size() > 1) return false;
    
40.         }
    
41.         return true;
    
42.     }
    
43. };

复制代码

糖逗