翻转二叉树

为什么80%的码农都做不了架构师？>>>   

思路：从根节点起，将每一个节点的左子树和右子树进行交换即可完成二叉树的翻转 。这是一个典型的可以使用递归解决的问题。

翻转二叉树的递归实现：

class TreeNode(object):
    def __init__(self, data):
        self.data = data
        self.left = None
        self.right = None

def 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

 

转载于:https://my.oschina.net/zhxx/blog/3039241