LeetCode 1361. 验证二叉树（图的出入度）

Michael阿明

发布于 2020-07-13 16:00:49

3210

发布于 2020-07-13 16:00:49

文章被收录于专栏：Michael阿明学习之路Michael阿明学习之路

1. 题目

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

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/validate-binary-tree-nodes 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

参考图的数据结构，出入度概念
二叉树可以看做特殊的图
记录每个节点的入度，root的入度为0，且root只有一个
其他节点的入度都只能为1
且总的入度等于 n-1

class Solution {
public:
    bool validateBinaryTreeNodes(int n, vector<int>& leftChild, vector<int>& rightChild) {
    	int indegree[n]  = {0};
    	bool findroot = false;
    	int i, sum = 0;
    	for(i = 0; i < n; ++i)
    	{
    		if(leftChild[i] != -1)
    			indegree[leftChild[i]]++;
    		if(rightChild[i] != -1)
    			indegree[rightChild[i]]++;	
    	}
    	for(i = 0; i < n; ++i)
    	{
    		if(!findroot && indegree[i]==0)
    			findroot = true;
    		else if(findroot && indegree[i]==0)
    			return false;
    		else if(indegree[i] > 1)
    			return false;
    		sum += indegree[i];
    	}
    	if(sum != n-1)
    		return false;
    	return true;
    }
};