翻转二叉树226

题目:翻转一棵二叉树。

输入:

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

输出:

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

思路:递归

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
class :
def __init__(self, x):
self.val = x
self.left = None
self.right = None


class Solution:
def invertTree(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""

if not root:
return root
# 递归过程,分别计算以左节点为根节点和以右节点为根节点的翻转后的二叉树
self.invertTree(root.left)
self.invertTree(root.right)
# 左右节点交换
root.left, root.right = root.right, root.left
return root

# 例子
tree = TreeNode(3)
left_tree = TreeNode(9)
right_tree = TreeNode(20)
right_tree.left = TreeNode(15)
right_tree.right = TreeNode(7)
tree.left = left_tree
tree.right = right_tree
print(Solution().invertTree(tree))