反转二叉树(python年度更新系列)
给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]
层序遍历,每次将其的孩子结点交换,如此反复即可。
class Solution(object):
def invertTree(self, root):
if root is None:
return
queue = [root]
while queue:
cur_node = queue.pop(0)
if cur_node.left:
queue.append(cur_node.left)
if cur_node.right:
queue.append(cur_node.right)
cur_node.left,cur_node.right = cur_node.right,cur_node.left
return root