力扣题目链接
翻转一棵二叉树。
输入:
输出:
可以发现,如何翻转一个二叉树?将每个父节点下的子节点互换就行了,4下面2分支树和7分支树互换,2下面1和3互换,7下面6和9互换。在前面可以自学一下关于二叉树的理论基础,二叉树的遍历形式上分为3种,前序遍历、中序遍历和后序遍历,分别对应父节点和子节点。前序遍历(中左右)、中序遍历(左中右)、后序遍历(右中左),三种遍历又分别可以用递归法和迭代法,可以挑一种作为自己精通掌握的,再了解其他方法做到融会贯通。
递归法前序遍历完整代码如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def invertTree(self, root: TreeNode) -> TreeNode:
if not root:
return None
root.left, root.right = root.right, root.left
self.invertTree(root.left)
self.invertTree(root.right)
return root
代码理解起来很简单,首先判断有没有父节点(根节点),没有了话就return到None。互换当前父节点下的左右节点,分别将左右节点利用invertTree函数进行递归,直到最后不存在下一个节点。根据相同的思路可以利用不同的方法编写代码,不理解的时候可以画一下图便于自己理解。