题目分析:
二叉搜索树中的两个节点被错误地交换。请在不改变其结构的情况下,恢复这棵树。
示例 1:
解题思路:
第一种:中序遍历将数值取出来,然后排序,然后再插入进入,使用的空间复杂度是O(n),见提交代码1。
第二种:使用指针找到变动的那两个值,空间复杂度为常量见提交代码2.
- 设置n1,n2的值用于寻找那两个点,
self.n1 = TreeNode(None); self.n2 = TreeNode(None)
- 设置pre节点用于保存上一个节点,根据上一个节点判断当前节点是否是错误的
self.pre = TreeNode(None)
- 中序遍历找到
root.val < self.pre.val
保存为n1,n2 - 交换n1,n2的值
第二种过程如下图:
很明显3和5颠倒了。那么在中序遍历时:当碰到第一个逆序时:为5->4,那么将n1指向5,n2指向4,注意,此时n1已经确定下来了。然后pre和root一直向后遍历,直到碰到第二个逆序时:4->3,此时将n2指向3,那么n1和n2都已经确定,只需要交换节点的值即可。pre指针用来比较中序遍历中相邻两个值的大小关系。
提交代码1:(递归+列表,Runtime: 80 ms, faster than 95.07% )
class Solution:
def inorder(self, root: TreeNode, list_num: list, list_root: list) -> None:
if root:
self.inorder(root.left, list_num, list_root)
list_num.append(root.val);list_root.append(root)
self.inorder(root.right, list_num, list_root)
def recoverTree(self, root: TreeNode) -> None:
list_nums = []
list_root = []
self.inorder(root, list_nums, list_root)
list_nums.sort()
for i, root in enumerate(list_root):
root.val = list_nums[i]
提交代码2:(递归+指针,Runtime: 80 ms, faster than 95.07% )
class Solution:
def inorder(self, root: TreeNode) -> None:
if root:
self.inorder(root.left)
if self.pre and root.val != None and self.pre.val != None and root.val < self.pre.val:
self.n2 = root
if self.n1.val == None: self.n1 = self.pre
self.pre = root
self.inorder(root.right)
def recoverTree(self, root: TreeNode) -> None:
self.n1 = TreeNode(None); self.n2 = TreeNode(None)
self.pre = TreeNode(None)
self.inorder(root)
self.n1.val, self.n2.val = self.n2.val, self.n1.val