二叉树问题：leetcode(572) 另一个树的子树

题目描述

给定两个非空二叉树 s 和 t，检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。
示例 1:
给定的树 s:

     5
    / \
   4   5
  / \
 1   2

给定的树 t：

   4 
  / \
 1   2

返回true，因为 t 与 s 的一个子树拥有相同的结构和节点值。

示例2
给定的树 s：

     3
    / \
   4   5
  / \
 1   2
    /
   0

给定的树 t：

   4
  / \
 1   2

返回false。

思路

用到了两个递归函数，isSameTree判断两课树是否相等，isSubtree判断树t是否是树s的子树，注意s为None时返回False

class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution(object):
    def isSubtree(self, s, t):
        """
        :type s: TreeNode
        :type t: TreeNode
        :rtype: bool
        """
        if s == None: return False
        if self.isSameTree(s, t): return True
        return self.isSubtree(s.left, t) or self.isSubtree(s.right, t)

    def isSameTree(self, s, t):
        if (s == None) != (t == None): return False
        if s == None: return True
        if s.val == t.val: return self.isSameTree(s.left, t.left) and self.isSameTree(s.right, t.right)
        else: return False