python——Leetcode 100. 相同的树

题目

给定两个二叉树，编写一个函数来检验它们是否相同。

如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

示例 1:

输入:       1         1
          / \       / \
         2   3     2   3

        [1,2,3],   [1,2,3]

输出: true
示例 2:

输入:      1          1
          /           \
         2             2

        [1,2],     [1,null,2]

输出: false
示例 3:

输入:       1         1
          / \       / \
         2   1     1   2

        [1,2,1],   [1,1,2]

输出: false

思路

思路类似于二叉树的遍历，首先判断两个是否为空的问题，再判断p的根节点和q的根节点是否相同，然后遍历左右节点。代码如下：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isSameTree(self, p, q):
        """
        :type p: TreeNode
        :type q: TreeNode
        :rtype: bool
        """
        # 1.首先判断p,q为空的情况
        if p==None and q==None:
            return True
        elif p==None and q!=None:
            return False
        elif p!=None and q==None:
            return False
        # 2.都不为空时，先判断根节点的值是否相同
        if p.val!=q.val:
            return False
        l = self.isSameTree(p.left, q.left)#再判断左子树是否相同，返回的值用l接受
        r = self.isSameTree(p.right, q.right)#最后判断右子树是否相同，返回的值用r接受
        # 注意：这种情况下，python使用函数递归时，是self.函数名（参数），如果原函数参数没有self，那递归调用和c语言一样
        return l and r