2022跟着leedcode学数据结构--第38天

100 阅读1分钟

「这是我参与2022首次更文挑战的第38天,活动详情查看:2022首次更文挑战

image.png

[剑指 Offer 26. 树的子结构]

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

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

例如: 给定的树 A:

image.png

给定的树 B:

  image.png

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

示例 1:

输入:A = [1,2,3], B = [3,1] 输出:false 示例 2:

输入:A = [3,4,5,1,2], B = [4,1] 输出:true 限制:

0 <= 节点个数 <= 10000

解题思路:

  • 不太会别的方法,还是中规中矩的DFS算法
  • 先判断B是否为空 为空返回True
  • 我们先写一个辅助方法DFS来帮助判断在是否A 包含B
  • 通过对主体的递归进行判断A是否包含B
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isSubStructure(self, A: TreeNode, B: TreeNode) -> bool:
        if not B: return False
        queue = [A]
        while queue:
            ans = queue.pop(0)
            if ans.val == B.val:
                if self.DFS(ans, B):
                    return True
            if ans.left:
                queue.append(ans.left)
            if ans.right:
                queue.append(ans.right)
        return False

    def DFS(self,A, B):
        queue = [(A, B)]
        while queue:
            nodeLeft, nodeRight = queue.pop(0)
            if not nodeLeft or nodeLeft.val != nodeRight.val:
                return False
            if nodeRight.left:
                queue.append((nodeLeft.left, nodeRight.left))
            if nodeRight.right:
                queue.append((nodeLeft.right, nodeRight.right))
        return True

执行结果:

image.png

[剑指 Offer 27. 二叉树的镜像]

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

例如输入:

image.png

镜像输出:

image.png

示例 1:

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

限制:

0 <= 节点个数 <= 1000

解题思路:

  • 这个题真的上来直接就是判断是否为空
  • 递归、交换就行了,实在没有什么其他方法了,就直接写吧
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

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

执行结果:

image.png