剑指offer-搜索与回溯算法II

最新推荐文章于 2024-09-18 18:23:55 发布

MGonster

最新推荐文章于 2024-09-18 18:23:55 发布

阅读量144

点赞数

分类专栏：剑指offer 文章标签：算法 leetcode

本文链接：https://blog.csdn.net/huihiu8yuih/article/details/123899239

版权

剑指offer 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

剑指 Offer 26. 树的子结构

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构，即 A中有出现和B相同的结构和节点值。

例如:
给定的树 A:

给定的树 B：

 4  
/ 
1

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

示例 1：

输入：A = [1,2,3], B = [3,1]
输出：false

示例 2：

输入：A = [3,4,5,1,2], B = [4,1]
输出：true

recur(A, B) 函数：

终止条件：

当节点 B 为空：说明树 B 已匹配完成（越过叶子节点），因此返回 true ；
当节点 A 为空：说明已经越过树 A 叶子节点，即匹配失败，返回 false ；
当节点 A 和 B 的值不同：说明匹配失败，返回 falsefalse ；

返回值：

判断 A 和 B 的左子节点是否相等，即 recur(A.left, B.left) ；
判断 A 和 B 的右子节点是否相等，即 recur(A.right, B.right) ；

isSubStructure(A, B) 函数：

特例处理： 当树 A 为空或树 B 为空时，直接返回 false ；
返回值： 若树 B 是树 A 的子结构，则必满足以下三种情况之一，因此用或 || 连接；

以节点 A 为根节点的子树包含树 B ，对应 recur(A, B)；
树 B 是树 A 左子树的子结构，对应 isSubStructure(A.left, B)；
树 B 是树 A 右子树的子结构，对应 isSubStructure(A.right, B)；

以上 2. 3. 实质上是在对树 A 做先序遍历。

class Solution:
    def isSubStructure(self, A: TreeNode, B: TreeNode) -> bool:
        def recur(A, B):
            if not B: return True
            if not A or A.val != B.val: return False
            return recur(A.left, B.left) and recur(A.right, B.right)

        return bool(A and B) and (recur(A, B) or self.isSubStructure(A.left, B) or self.isSubStructure(A.right, B))

剑指 Offer 27. 二叉树的镜像（简单）

请完成一个函数，输入一个二叉树，该函数输出它的镜像。

例如输入：

     4
   /   \
  2     7
 / \   / \
1   3 6   9

镜像输出：

     4
   /   \
  7     2
 / \   / \
9   6 3   1

示例 1：

输入：root = [4,2,7,1,3,6,9]
输出：[4,7,2,9,6,3,1]

方法一：递归法

二叉树镜像定义：对于二叉树中任意节点 root ，设其左 / 右子节点分别为 left, right ；则在二叉树的镜像中的对应 root 节点，其左 / 右子节点分别为 right, left 。

终止条件： 当节点 root 为空时（即越过叶节点），则返回 null ；
递推工作：
初始化节点 tmp ，用于暂存 root 的左子节点；
开启递归 右子节点 mirrorTree(root.right)，并将返回值作为 root 的 左子节点 。
开启递归 左子节点 mirrorTree(tmp) ，并将返回值作为 root 的 右子节点 。
返回值： 返回当前节点 root；

class Solution:
    def mirrorTree(self, root: TreeNode) -> TreeNode:
        if not root: return
        root.left, root.right = self.mirrorTree(root.right), self.mirrorTree(root.left)
        return root

方法二：辅助栈（或队列）
利用栈（或队列）遍历树的所有节点 nodenode ，并交换每个 nodenode 的左 / 右子节点。
算法流程：

特例处理： 当 root 为空时，直接返回 null ；
初始化： 栈（或队列），本文用栈，并加入根节点 root 。
循环交换： 当栈 stack 为空时跳出；
- 出栈：记为 node ；
- 添加子节点：将 node 左和右子节点入栈；
- 交换：交换 node 的左 / 右子节点。
返回值： 返回根节点 root。

class Solution:
    def mirrorTree(self, root:TreeNode) ->TreeNode:
        if not root: return
        stack = [root]
        while stack:
            node = stack.pop()
            if node.left: stack.append(node.left)
            if node.right: stack.append(ndoe.right)
            node.left, node.right = node.right, node.left
        return root

剑指 Offer 28. 对称的二叉树

请实现一个函数，用来判断一棵二叉树是不是对称的。如果一棵二叉树和它的镜像一样，那么它是对称的。

例如，二叉树 [1,2,2,3,4,4,3] 是对称的。

但是下面这个 [1,2,2,null,3,null,3] 则不是镜像对称的:

示例 1：

输入：root = [1,2,2,3,4,4,3]
输出：true

示例 2：

输入：root = [1,2,2,null,3,null,3]
输出：false

算法流程：
isSymmetric(root) ：

特例处理：若根节点 root 为空，则直接返回 true 。
返回值：即 recur(root.left, root.right) ;

recur(L, R) ：

终止条件：
- 当 L 和 R 同时越过叶节点：此树从顶至底的节点都对称，因此返回 true ；
- 当 L 或 R 中只有一个越过叶节点：此树不对称，因此返回 false ；
- 当节点 L 值 != 节点 R 值：此树不对称，因此返回 false ；
递推工作：
- 判断两节点 L.left 和 R.right是否对称，即 recur(L.left, R.right) ；
- 判断两节点 L.right 和 R.left 是否对称，即 recur(L.right, R.left) ；
返回值：两对节点都对称时，才是对称树，因此用与逻辑符 && 连接。

class Solution:
    def isSymmetric(self,root:TreeNode) -> bool:
        def recur(L, R):
            if not L and not R: return True
            if not L or not R or L.val != R.val: return False
            return recur(L.left, R.right) and recur(L.right, R.left) 
        return recur(root.left, root.right) if root else True