leetcode99 恢复二叉搜索树

99. 恢复二叉搜索树

难度中等634

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

示例 1：

输入：root = [1,3,null,null,2]
输出：[3,1,null,null,2]
解释：3 不能是 1 左孩子，因为 3 > 1 。交换 1 和 3 使二叉搜索树有效。

这一题乍一看可能有点复杂，实际上非常简单，我们需要做的就是通过中序遍历来将节点都储存在一个数组中。这时候本题就被转化成了恢复一个两个值被误换的增序数组的问题，而交换直接交换两个节点的值即可。

恢复数组顺序的方法：我们将数列中第一个有问题的值的位置标记为x，后一个位置标记为y

1. 1 4 3 5 2

        这种情况下4和2不连续，可以发现4比它下一位要大，2比其前一位要小，而3也是比二小的，所以我们可以先通过给x赋最大值的方式对它进行标记，只有x在没有存储数列中的值的时候才能给x赋值，而只有再给x赋值之后才能对y赋值

2. 1 3 2 4 5

        这种情况下3和2是连续的，后面没有符合当前位小于前一位的情况了，所以当我们遍历到数组末尾的时候发现y还没被赋值，就是这种情况，直接令y = x + 1；

class Solution {
    List<TreeNode> list = new ArrayList<>();
    public void recoverTree(TreeNode root) {
        //先通过中序遍历将节点按顺序存入list
        dfs(root);
        int x = Integer.MAX_VALUE, y = Integer.MAX_VALUE;
        for(int i = 0; i < list.size() - 1; i++){
            //因为数组是升序的，所以如果当前值大于后面节点的值，该值或者后面的值是有问题的
            if(list.get(i).val > list.get(i + 1).val){
                //因为x被赋了最大值，这一步判断x是否已经存储了节点
                if(x > i) x = i;
                else {
                    //先判断x是否赋值，x若已经被赋值则赋予y，因为问题节点只有两个，所以直接退出循环
                    y = i + 1;
                    break;
                }
            }
            //x 和 x+1 交换的情况
            if(y > i && i + 1 == list.size() - 1) y = x + 1;
        }
        swap(list.get(x), list.get(y));
    }
    //中序遍历
    public void dfs(TreeNode root){
        if(root == null) return;
        dfs(root.left);
        list.add(root);
        dfs(root.right);
    }
    //交换两个节点的值
    public void swap(TreeNode a, TreeNode b){
        TreeNode temp = new TreeNode();
        temp.val = a.val;
        a.val = b.val;
        b.val = temp.val;
    }
}