剑指 Offer 27. 二叉树的镜像https://leetcode.cn/problems/er-cha-shu-de-jing-xiang-lcof/
二叉树镜像定义: 对于二叉树中任意节点 root,设其左 / 右子节点分别为 left, right;则在二叉树的镜像中的对应 root节点,其左 / 右子节点分别为 right, left。
即,对任意节点,交换它的左右邻居。
方法一:递归法
- 根据二叉树镜像的定义,考虑递归遍历(dfs)二叉树,交换每个节点的左 / 右子节点,即可生成二叉树的镜像。
终止条件: 当节点 root为空时(即越过叶节点,叶节点的左右节点为空),则返回 null ;
递推工作:
初始化节点 tmp,用于暂存 root的左子节点;
开启递归 右子节点 mirrorTree(root.right) ,并将返回值作为 root 的 左子节点 。
开启递归 左子节点 mirrorTree(tmp) ,并将返回值作为 root的 右子节点 。
返回值: 返回当前节点 root;
class Solution:
def mirrorTree(self, root):
if not root: return None
tmp=root.left
root.left=self.mirrorTree(root.right)
root.right=self.mirrorTree(tmp)
return root
递归的本质是栈,也可以用栈(列表就可以表示)
class Solution:
def mirrorTree(self, root):
if not root: return
stack=[]
stack.append(root)
while stack:
node=stack.pop()
node.left,node.right=node.right,node.left
if node.left:
stack.append(node.left)
if node.right:
stack.append(node.right)
return root