LeetCode 力扣官方题解 | 965. 单值二叉树
- 单值二叉树
题目描述
难易度:简单
如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。
只有给定的树是单值二叉树时,才返回 true;否则返回 false。
示例 1:
输入:[1,1,1,1,1,null,1]
输出:true
示例 2 :
输入:[2,2,2,5,2]
输出:false
提示:
1.给定树的节点数范围是 [1, 100 ] 。
2.每个节点的值都是整数,范围为 [0, 99 ] 。
解决方案
方法一:深度优先搜索
思路和算法
一棵树的所有节点都有相同的值,当且仅当对于树上的每一条边的两个端点,它们都有相同的值(这样根据传递性,所有节点都有相同的值)。
因此,我们可以对树进行一次深度优先搜索。当搜索到节点 x 时,我们检查 x 与 x 的每一个子节点之间的边是否满足要求。例如对于左子节点而言,如果其存在并且值与 x 相同,那么我们继续向下搜索该左子节点;如果值与 x 不同,那么我们直接返回 False 。
代码
C++
class Solution {
public:
bool isUnivalTree(TreeNode* root) {
if (!root) {
return true;
}
if (root->left) {
if (root->val != root->left->val || !isUnivalTree(root->left)) {
return false;
}
}
if (root->right) {
if (root->val != root->right->val || !isUnivalTree(root->right)) {
return false;
}
}
return true;
}
};
Java
class Solution {
public boolean isUnivalTree(TreeNode root) {
if (root == null) {
return true;
}
if (root.left != null) {
if (root.val != root.left.val || !isUnivalTree(root.left)) {
return false;
}
}
if (root.right != null) {
if (root.val != root.right.val || !isUnivalTree(root.right)) {
return false;
}
}
return true;
}
}
C#
public class Solution {
public bool IsUnivalTree(TreeNode root) {
if (root == null) {
return true;
}
if (root.left != null) {
if (root.val != root.left.val || !IsUnivalTree(root.left)) {
return false;
}
}
if (root.right != null) {
if (root.val != root.right.val || !IsUnivalTree(root.right)) {
return false;
}
}
return true;
}
}
Python3
class Solution:
def isUnivalTree(self, root: TreeNode) -> bool:
if not root:
return True
if root.left:
if root.val != root.left.val or not self.isUnivalTree(root.left):
return False
if root.right:
if root.val != root.right.val or not self.isUnivalTree(root.right):
return False
return True
C
bool isUnivalTree(struct TreeNode* root){
if (!root) {
return true;
}
if (root->left) {
if (root->val != root->left->val || !isUnivalTree(root->left)) {
return false;
}
}
if (root->right) {
if (root->val != root->right->val || !isUnivalTree(root->right)) {
return false;
}
}
return true;
}
JavaScript
var isUnivalTree = function(root) {
if (!root) {
return true;
}
if (root.left) {
if (root.val !== root.left.val || !isUnivalTree(root.left)) {
return false;
}
}
if (root.right) {
if (root.val !== root.right.val || !isUnivalTree(root.right)) {
return false;
}
}
return true;
};
Golang
func isUnivalTree(root *TreeNode) bool {
return root == nil || (root.Left == nil || root.Val == root.Left.Val && isUnivalTree(root.Left)) &&
(root.Right == nil || root.Val == root.Right.Val && isUnivalTree(root.Right))
}
复杂度分析
时间复杂度:O(n) ,其中 n 是二叉树的节点个数。我们遍历二叉树的每个节点至多一次。
空间复杂度:O(n) ,即为深度优先搜索中需要使用的栈空间。
获取更多详情:LeetCode 力扣官方题解 | 965. 单值二叉树
编辑/版式:pingping