leetcode地址：恢复二叉搜索树
给你二叉搜索树的根节点 root ，该树中的 恰好 两个节点的值被错误地交换。请在不改变其结构的情况下，恢复这棵树 。
示例 1：
输入：root = [1,3,null,null,2]
输出：[3,1,null,null,2]
解释：3 不能是 1 的左孩子，因为 3 > 1 。交换 1 和 3 使二叉搜索树有效。
示例 2：
输入：root = [3,1,4,null,null,2]
输出：[2,1,4,null,null,3]
解释：2 不能在 3 的右子树中，因为 2 < 3 。交换 2 和 3 使二叉搜索树有效。
提示：
树上节点的数目在范围 [2, 1000] 内
-231 <= Node.val <= 231 - 1
实现思路
这个问题要求恢复一个二叉搜索树，其中恰好有两个节点的值被错误地交换了，但是不改变其结构。二叉搜索树的特性是左子树的所有节点小于根节点，右子树的所有节点大于根节点。
中序遍历
二叉搜索树进行中序遍历得到的序列应该是递增的。如果有两个节点交换了位置，会导致中序遍历序列中出现一对不满足递增关系的节点。
寻找错误节点
在中序遍历过程中，记录前驱节点和当前节点。
如果出现前驱节点的值大于当前节点的值，则这两个节点是需要交换的节点。
恢复节点
如果发现了需要交换的节点，记录下来。
最后交换这两个节点的值，使得树恢复为二叉搜索树的结构。
代码详解
# Definition for a binary tree node.
class TreeNode:def __init__(self, val=0, left=None, right=None):self.val = valself.left = leftself.right = rightdef recoverTree(root: TreeNode) -> None:x = y = prev = Nonedef inorder(node):nonlocal x, y, previf not node:returninorder(node.left)if prev and prev.val >= node.val:if x is None:x = prevy = nodeprev = nodeinorder(node.right)inorder(root)x.val, y.val = y.val, x.val# Helper function to print inorder traversal
def print_inorder(root):if not root:returnprint_inorder(root.left)print(root.val, end=" ")print_inorder(root.right)# Example usage:
# Example 1: [1,3,null,null,2]
root1 = TreeNode(1)
root1.right = TreeNode(3)
root1.right.left = TreeNode(2)print("Before recovery:")
print_inorder(root1)
print()recoverTree(root1)print("After recovery:")
print_inorder(root1)go实现
package mainimport "fmt"type TreeNode struct {Val intLeft *TreeNodeRight *TreeNode
}func recoverTree(root *TreeNode) {var x, y, prev *TreeNodevar inorder func(*TreeNode)inorder = func(node *TreeNode) {if node == nil {return}inorder(node.Left)if prev != nil && prev.Val >= node.Val {if x == nil {x = prev}y = node}prev = nodeinorder(node.Right)}inorder(root)x.Val, y.Val = y.Val, x.Val
}// Helper function to print inorder traversal
func printInorder(root *TreeNode) {if root == nil {return}printInorder(root.Left)fmt.Printf("%d ", root.Val)printInorder(root.Right)
}func main() {// Example 1: [1,3,null,null,2]root := &TreeNode{Val: 1}root.Right = &TreeNode{Val: 3}root.Right.Left = &TreeNode{Val: 2}fmt.Println("Before recovery:")printInorder(root)fmt.Println()recoverTree(root)fmt.Println("After recovery:")printInorder(root)
}kotlin实现
class TreeNode(var `val`: Int) {var left: TreeNode? = nullvar right: TreeNode? = null
}fun recoverTree(root: TreeNode?) {var x: TreeNode? = nullvar y: TreeNode? = nullvar prev: TreeNode? = nullfun inorder(node: TreeNode?) {if (node == null) returninorder(node.left)if (prev != null && prev!!.`val` >= node.`val`) {if (x == null) {x = prev}y = node}prev = nodeinorder(node.right)}inorder(root)// Swap the values of x and yval temp = x!!.`val`x!!.`val` = y!!.`val`y!!.`val` = temp
}// Helper function to print the tree in-order
fun printInOrder(node: TreeNode?) {if (node == null) returnprintInOrder(node.left)print("${node.`val`} ")printInOrder(node.right)
}fun main() {// Example 1: [1,3,null,null,2]val root1 = TreeNode(1)root1.right = TreeNode(3)root1.right!!.left = TreeNode(2)println("Before recovery:")printInOrder(root1)println()recoverTree(root1)println("After recovery:")printInOrder(root1)println()
}
