Leetcode 99. 恢复二叉搜索树（Medium）-优快云博客

给你二叉搜索树的根节点 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

思路：就是找到两个要交换的节点进行交换，交换的节点在他的上下文中必然是错误的，什么意思呢，就是破坏了二叉搜索树的性质，左小，右大。把这个二叉树用中序遍历抽象成链表，应该是从小到大递增的，找出两个故障的节点，然后进行交换。怎么找出来呢，就和链表一样，使用连续的前后两个节点，正常情况是前<后，然后当前>后，可以进行交换，然后一直交换到最后结束。或者也可以给他标识起来，前节点总是固定的，和冒泡排序类似，后节点就一直变，等最后进行一次交换就OK了。

/**
 * 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 first = null;
    TreeNode second = null;
    TreeNode prev = new TreeNode(Integer.MIN_VALUE);

    public void recoverTree(TreeNode root) {
        findAnswer(root);
        // 进行交换
        int temp = first.val;
        first.val = second.val;
        second.val = temp;
    }

    public void findAnswer(TreeNode root) {
        if(root == null) return;

        findAnswer(root.left);
        //当前节点不是最小的，需要交换
        if(prev.val > root.val) {
            if(first == null) {
                first = prev;
            }
            //需要把不符合条件的最小的值根不符合条件的最大的值交换
            second = root;
        }
        // prev代表的是上一个节点
        prev = root;

        findAnswer(root.right);
    }
}