leetcode 100. 相同的树 DFS练习

DFS练习

错误代码1

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

class Solution:
    def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
        if p.val==q.val:
            self.isSameTree(p.left,q.left)
            self.isSameTree(p.right,q.right)
        else:
            return False


#报错
# if p.val==q.val:
# AttributeError: 'NoneType' object has no attribute 'val'

测试以下，p、q又是成功创建了的：

print(q.val)
# 返回值：1

那为什么会报错啊？

调试后明白，当运行到最后一个递归后，为空了，没有设定退出条件。

错误代码2

错误原因，退出条件没有设对。

class Solution:
    def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
        if not p and q:
            return False
        elif not q and p:
            return False
        elif not p and not q:
            return True
        # 可以第一个设置为not and not，第二个设置为 not or not，就把前三个条件统一了。
        # 其实还差一个关键条件。
        # 前面只是说对为空的判断，但是如果val不相等就没有判断。
        else:
            return self.isSameTree(p.left,q.left) and self.isSameTree(p.right,q.right)

 参考代码

class Solution:
    def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
        if not p and not q :
            return True
        elif not p or not q:
            return False
        elif p.val!=q.val:
            return False
        else:
            return self.isSameTree(p.left,q.left) and self.isSameTree(p.right,q.right)