【Leetcode】1361. Validate Binary Tree Nodes

题目地址：

https://leetcode.com/problems/validate-binary-tree-nodes/

给定一个有向图，并且每个顶点的出度不超过$2$，判断其是否是二叉树。

它是二叉树的条件是以下这么几个：
1、其作为无向图是没有环的；
2、其作为无向图是连通的；
3、只有一个入度为$0$的点。

必要性显然。充分性可以用数学归纳法来做。如果图只有一个顶点，显然。如果图有$n$个顶点，首先找到那个入度为$0$的顶点$r$，由于图连通，所以这个顶点一定有出边，取其出边指向的顶点$u$，如果$u$的入度不为$1$，那么顺着$u$的入边向上找，总能找到另一个入度为$0$的点，这样就矛盾了。所以$u$的入度一定为$1$。现在将这条边断开，由于图连通，所以就形成了两个连通分支，含$u$的那个仍然满足三个条件，由归纳假设知道，其是一棵二叉树。这样就证明了$r$的左右顶点都是二叉树的树根，证明结束。

可以用DFS来做，先找到入度为$0$的点，如果找不到或者找到多个，则不是二叉树。否则从树根开始进行DFS，如果发现走到了重复点，则不是二叉树。如果能遍历到所有点，则是二叉树。代码如下：

class Solution {
 public:
  bool validateBinaryTreeNodes(int n, vector<int>& left, vector<int>& right) {
    int ind[n];
    memset(ind, 0, sizeof ind);
    for (int i = 0; i < n; i++) {
      if (~left[i]) ind[left[i]]++;
      if (~right[i]) ind[right[i]]++;
    }

    int root = -1;
    for (int i = 0; i < n; i++) {
      if (!ind[i])
        if (root == -1)
          root = i;
        else
          return false;
    }

    if (root == -1) return false;

    vector<bool> vis(n, false);
    int cnt = 0;
    return dfs(root, vis, left, right, cnt) && cnt == n;
  }

  bool dfs(int u, vector<bool>& vis, vector<int>& left, vector<int>& right,
           int& cnt) {
    vis[u] = true;
    cnt++;
    if (~left[u] && (vis[left[u]] || !dfs(left[u], vis, left, right, cnt)))
      return false;
    if (~right[u] && (vis[right[u]] || !dfs(right[u], vis, left, right, cnt)))
      return false;
    return true;
  }
};

时空复杂度$O(n)$。