携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第4天,点击查看活动详情
一、题目描述:
100. 相同的树 - 力扣(LeetCode) (leetcode-cn.com)
给你两棵二叉树的根节点 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
提示:
- 两棵树上的节点数目都在范围 [0, 100] 内
- -10^4 <= Node.val <= 10^4
二、思路分析:
递归大法好
对比两棵树的节点指针p和q:
-
两棵树不同有两种情况:
- 两棵树的结构不同,即结点的位置不同
- 两棵树的数值不同,即结点的数值不同
-
两棵树相同只有一种情况:结点位置相同并且结点的值相同
三、AC 代码:
# 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: TreeNode, q: TreeNode) -> bool:
if not p and not q: # 递归终止条件:两个需要对比的结点都为空,即已经到树的末梢了
return True
if p is None or q is None: # 两棵树不同的情况1:两棵树中只有一棵树有左孩子结点或者只有一棵树有右孩子结点
return False
if p.val != q.val: # 两棵树不同的情况2:对比的两个结点的值不同
return False
else:
return self.isSameTree(p.left, q.left) and self.isSameTree(p.right, q.right) # 递归处理
范文参考:
【仗剑骑士】C++, 二叉树的套路, 速度超过100%的用户+ 100. 相同的树 - 相同的树 - 力扣(LeetCode)