【Leetcode】python - [572] Subtree of Another Tree 個人解法筆記

題目出處

572. Subtree of Another Tree

難度

easy

個人範例程式碼

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def isSubtree(self, root: Optional[TreeNode], subRoot: Optional[TreeNode]) -> bool:
        # end of recursion
        if not root:
            return False

        # define
        if self.is_same_tree(root, subRoot):
            return True

        # split
        return self.isSubtree(root.left, subRoot) or self.isSubtree(root.right, subRoot)

    def is_same_tree(self, root, subroot):
        # end of recursion
        if not root and not subroot:
            return True

        if not root or not subroot: # only one side left
            return False

        if root.val != subroot.val: 
            return False

        return self.is_same_tree(root.left, subroot.left) and self.is_same_tree(root.right, subroot.right)

算法說明

原本的 is_sub_tree,就已經可以幫助我們把大題目拆成小題目了,
我們只需要再多補上一個,比較兩個 tree 是否相同的函數即可。

input handling

if not root,return False (上層只要有一次 True 即可,因此用 or 比較其他答案)

Boundary conditions

用 not root,來判斷 Tree 的結尾。

Reference

Licensed under CC BY-NC-SA 4.0