思路:从根节点起,将每一个节点的左子树和右子树进行交换即可完成二叉树的翻转 。这是一个典型的可以使用递归解决的问题。
翻转二叉树的递归实现:
class TreeNode(object): def __init__(self, data): self.data = data self.left = None self.right = Nonedef invert_tree(node): if node is None: return None node.left, node.right = invert_tree(node.right), invert_tree(node.left) return node
用递归可以解决的问题,同样使用迭代也可以解决 。
迭代时可以使用队列, 从根节点开始入栈,出栈,交换左右子树,然后左子树和右子树再分别入栈 ,这样就可以以层次遍历的方式翻转二叉树了 。
代码实现:
def invert_tree(node): if node is None: return None queue = collections.deque() if node: queue.append(node) while queue: n = queue.popleft() if n.left: queue.append(n.left) if n.right: queue.append(n.right) n.left, n.right = n.right, n.left return node