leetcode刷题题解——99. 恢复二叉搜索树

public void recoverTree(TreeNode root) {
    Stack<TreeNode> stack = new Stack<>();
    TreeNode node = root;
    TreeNode prev = null;
    TreeNode node1=null,node2=null;
    while (!stack.isEmpty()||node!=null){
        while (node!=null){
            stack.push(node);
            node = node.left;
        }
        node = stack.pop();
        if (prev!=null&&node.val<prev.val){
            if (node1==null){
                node1 = prev;
                node2 = node;
            }else {
                node2 = node;
            }
        }
        prev = node;
        node = node.right;
    }
    if (node1!=null&&node2!=null){
        int temp = node1.val;
        node1.val = node2.val;
        node2.val = temp;
    }
}

思路：
题目说到交换节点，那么其实就只有两种可能，交换的节点相邻，或交换的节点不相邻
交换的节点相邻：
a_i和a_i+1交换，那么就出现了a_i>a_i+1保存两个节点
这个时候是不是就可以直接交换两个节点了呢？并不是，因为还有一种情况
a_i和a_j交换，交换后
a_i>a_i+1，a_j-1>a_j
发现这两种情况在第一个节点处，情况是一样的那么我们可以先保存第一种情况的两个节点，然后如果出现下一个不符合条件的点时将a_i+1用a_j替换
然后同一在循环结束后交换两个节点