1361. 验证二叉树

二叉树上有 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

这道题的思路是：

1.先找出根节点，且如果有多个根节点或没有根节点（两个点双向连接的情况），返回false；

2.利用bfs队列，从上到下从左到右，经过的点做标记，防止之后的结点向上连接。

3.对所有结点遍历过一遍之后，如果所有结点都被标记，证明符合题意，返回true，否则返回false；

#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
class Solution {
public:
    bool validateBinaryTreeNodes(int n, vector<int>& leftChild, vector<int>& rightChild) {
        vector<int> inDegree(n,0);
        for(int i=0;i<n;i++){
            if(leftChild[i]!=-1) inDegree[leftChild[i]]++;
            if(rightChild[i]!=-1) inDegree[rightChild[i]]++;
        }
        int root = -1;
        for(int i=0;i<n;i++){
            if(inDegree[i]!=0)continue;
            if(root != -1)return false;
            root = i;
        }
        if(root == -1 ) return false;
        queue<int> que;
        vector<bool> vi(n,false);
        vi[root]=true;
        que.push(root);
        while(!que.empty()){
            int q = que.front();
            que.pop();
            if(leftChild[q]!=-1){
                if(vi[leftChild[q]]==true) return false;
                vi[leftChild[q]]=true;
                que.push(leftChild[q]);
            }
            if(rightChild[q]!=-1){
                if(vi[rightChild[q]]==true) return false;
                vi[rightChild[q]]=true;
                que.push(rightChild[q]);
            }
        }
        return all_of(vi.begin(),vi.end(),[](const int &a){return a==1;});
    }
};
int main(){
    //freopen("D:\\YJ.txt","r",stdin);
    int n;
    cin>>n;
    vector<int> leftChild(n,0),rightChild(n,0);
    for(int i=0;i<n;i++)cin>>leftChild[i];
    for(int i=0;i<n;i++)cin>>rightChild[i];
    Solution solution;
    bool res = solution.validateBinaryTreeNodes(n, leftChild, rightChild);
    cout<<res;
    return 0;
    
}