99. 恢复二叉搜索树

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

题目描述

给你二叉搜索树的根节点 root ，该树中的两个节点被错误地交换。请在不改变其结构的情况下，恢复这棵树。

进阶：使用 O(n) 空间复杂度的解法很容易实现。你能想出一个只使用常数空间的解决方案吗？

示例 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 使二叉搜索树有效。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/recover-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

解题过程

解题思路

因为中序遍历就是二分搜索树按照从小到大排序的顺序，所以采用中序遍历处理。以示例2为例，按照中序遍历搞这棵树，用pre记录上一个节点，则会有下图这样的关系：

需要注意的是，上述的例子两个不符合规则的节点在中序遍历中是相邻的关系，而如果是两个不相邻的节点，则mistake1和mistake2的处理就需要有些许变化，比如这样的例子：4->2->3->1

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    TreeNode mistake1 = null;
    TreeNode mistake2 = null;
    TreeNode pre = null;
    public void recoverTree(TreeNode root) {

        recur(root);

        if(mistake1 != null && mistake2 != null){
            int temp = mistake1.val;
            mistake1.val = mistake2.val;
            mistake2.val = temp;
        }

    }

    public void recur(TreeNode root){
        if(root == null){
            return;
        }
        if(root.left != null){
            recur(root.left);
        }
        //中间处理逻辑
        if(pre != null && root.val < pre.val){
            if(mistake1 == null){//按照二叉搜索树的排序规则相邻的两个节点正好交换就是这样
                mistake1 = pre;
                mistake2 = root;
            }else{              //到这里表示按照规则，并非是两个相邻的节点交换，第一个要交换的节点上次记录了，这次记录第二个要交换的节点
                mistake2 = root;
            }
        }
        pre = root;
        
        if(root.right != null){
            recur(root.right);
        }

    }
}

总结