题目描述:
请完成一个函数,输入一个二叉树,该函数输出它的镜像,即交换树的所有左右子节点。
解决方案:辅助栈
- 使用栈的数据结构来辅助树进行镜像的完成
- 首先在栈中加入根节点(节点为空,返回Null),将节点出栈
- 其次,将节点的左右子节点入栈
- 然后,将左右节点交换
- 依次循环以上步骤,直到栈空时,退出,返回根节点(镜像完成)
解决代码如下:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def mirrorTree(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
if not root: return
stack = [root]
while stack:
node = stack.pop()
if node.left: stack.append(node.left)
if node.right: stack.append(node.right)
node.left, node.right = node.right, node.left
return root