2236. 判断根结点是否等于子结点之和

Problem#

给你一个二叉树根结点，判断该根结点是否等于它的左右子结点之和。

树只包含根结点、左子结点和右子结点。

https://leetcode.cn/problems/root-equals-sum-of-children/description/

示例 1：

输入：root = [10, 4, 6] 输出：true 解释：10 == 4 + 6，返回 true。

示例 2：

输入：root = [5, 3, 1] 输出：false 解释：5 != 3 + 1，返回 false。

约束：

树只包含 1 到 3 个结点
每个结点值在 [0, 100] 范围内

思路#

直接判断 root.val == root.left.val + root.right.val。注意题目保证左右子结点都存在。

1
# Definition for a binary tree node.
2
# class TreeNode:
3
#     def __init__(self, val=0, left=None, right=None):
4
#         self.val = val
5
#         self.left = left
6
#         self.right = right
7

8
class Solution:
9
    def checkTree(self, root: TreeNode) -> bool:
10
        return root.val == root.left.val + root.right.val

1
import unittest
2

3

4
class TreeNode:
5
    def __init__(self, val=0, left=None, right=None):
6
        self.val = val
7
        self.left = left
8
        self.right = right
9

10

11
class TestCheckTree(unittest.TestCase):
12
    def setUp(self):
13
        self.solution = Solution()
14

15
    def test_equal(self):
16
        root = TreeNode(10, TreeNode(4), TreeNode(6))
17
        self.assertTrue(self.solution.checkTree(root))
18

19
    def test_not_equal(self):
20
        root = TreeNode(5, TreeNode(3), TreeNode(1))
21
        self.assertFalse(self.solution.checkTree(root))
22

23
    def test_zero(self):
24
        root = TreeNode(0, TreeNode(0), TreeNode(0))
25
        self.assertTrue(self.solution.checkTree(root))
26

27

28
if __name__ == '__main__':
29
    unittest.main()