100. 相同的树python

问题描述

给你两棵二叉树的根节点 p 和 q ，编写一个函数来检验这两棵树是否相同。

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

示例 1:

输入：p = [1,2,3], q = [1,2,3]
输出：true

示例 2:

输入：p = [1,2], q = [1,null,2]
输出：false

示例 3:

输入：p = [1,2,1], q = [1,1,2]
输出：false

解答

class Solution(object):
    def isSameTree(self, p, q):
        """
        :type p: TreeNode
        :type q: TreeNode
        :rtype: bool
        """
        if p is None and q is None:
            return True
        if p is None or q is None:
            return False

        if p.val == q.val:
            flag = self.isSameTree(p.left, q.left)
            if not flag:
                return False
            flag = self.isSameTree(p.right, q.right)
            if not flag:
                return False
        else:
            return False
        return True
        

运行结果

心得

注意不满足条件就停止