题目地址:
https://leetcode.com/problems/validate-binary-tree-nodes/
给定一个有向图,并且每个顶点的出度不超过 2 2 2,判断其是否是二叉树。
它是二叉树的条件是以下这么几个:
1、其作为无向图是没有环的;
2、其作为无向图是连通的;
3、只有一个入度为
0
0
0的点。
必要性显然。充分性可以用数学归纳法来做。如果图只有一个顶点,显然。如果图有 n n n个顶点,首先找到那个入度为 0 0 0的顶点 r r r,由于图连通,所以这个顶点一定有出边,取其出边指向的顶点 u u u,如果 u u u的入度不为 1 1 1,那么顺着 u u u的入边向上找,总能找到另一个入度为 0 0 0的点,这样就矛盾了。所以 u u u的入度一定为 1 1 1。现在将这条边断开,由于图连通,所以就形成了两个连通分支,含 u u u的那个仍然满足三个条件,由归纳假设知道,其是一棵二叉树。这样就证明了 r r r的左右顶点都是二叉树的树根,证明结束。
可以用DFS来做,先找到入度为 0 0 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 ) O(n) O(n)。