LeetCode----Invert Binary Tree

Invert Binary Tree

Invert a binary tree.

4
/   \
2     7
/ \   / \
1   3 6   9

to

4
/   \
7     2
/ \   / \
9   6 3   1

Trivia:
This problem was inspired by this
original tweet by Max
Howell:

Google: 90% of our engineers use the software you wrote (Homebrew), but you can’t invert a binary tree on a whiteboard so fuck off.

分析：

这题是谷歌面试写出OS X中写出Homebrew软件的程序员的问题，结果他没有写出来，然后就悲剧了。我用递归和非递归都实现了一遍，然而并不代表我能去谷歌。

递归中，只需要递归地将当前节点的左右子节点交换即可。

非递归中，需要使用栈保存当前节点左右非空子节点，然后在循环中弹栈访问，并交换其左右子节点。

递归代码：

class Solution(object):
def invertTree(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
if not root:
return None
r = root.right
l = root.left
root.right = self.invertTree(l)
root.left = self.invertTree(r)
return root

非递归代码：

class Solution(object):
def invertTree(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
if not root:
return None
st = []
p = root
st.append(p)
while st:
p = st.pop()
l = p.left
r = p.right
if l or r:
p.left = r
p.right = l
if l:
st.append(l)
if r:
st.append(r)
return root