LeetCode 1361.验证二叉树

欢迎来访

分析

若是二叉树应该满足下面3个条件：
1.有且仅有一个根节点(入度为0)。
2.其他的节点的入度均为1。
3.所有的节点均在一棵树上(其实这个就相当于条件1了)。

c++代码

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;
        // 要明确的是跳出这个循环有两种可能性：1.找到root 2.遍历完所有的点
        for (int i = 0; i < n; i++){
            if (indegree[i] == 0){
                root = i;
                break;
            }
        }
        if (root == -1) return false;

        //cout << "root:" << root << endl;
        
        vector<bool> vis(n, false);
        queue<int> q;
        q.push(root);
        vis[root] = true;
        while (q.size()){
            int t = q.front();
            q.pop();
            if (leftChild[t] != -1){
                if (vis[leftChild[t]]) return false;
                else{
                    q.push(leftChild[t]);
                    vis[leftChild[t]] = true;
                }
            }
            if (rightChild[t] != -1){
                if (vis[rightChild[t]]) return false;
                else{
                    q.push(rightChild[t]);
                    vis[rightChild[t]] = true;
                }
            }
        }
        //cout << "here" << endl;
        for (int i = 0; i < n; i++){
            if (!vis[i]) {
                //cout << i << endl;
                return false;
            }
        }
        return true;
    }
};