同剑指offer28题:二叉树的镜像
思路
不管左右子节点存在与否,直接替换就好
代码
class Solution(object):
def invertTree(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
if not root:
return None
#不需要进行左右子节点是否存在的判断
root.left,root.right = root.right,root.left
self.invertTree(root.left)
self.invertTree(root.right)
return root