[LeetCode] 99. Recover Binary Search Tree(复原BST) ☆☆☆☆☆

Recover Binary Search Tree leetcode java

https://leetcode.com/problems/recover-binary-search-tree/discuss/32535/No-Fancy-Algorithm-just-Simple-and-Powerful-In-Order-Traversal

描述

解析

解决方法是利用中序遍历找顺序不对的两个点，最后swap一下就好。

因为这中间的错误是两个点进行了交换，所以就是大的跑前面来了，小的跑后面去了。

所以在中序遍利时，遇见的第一个顺序为递减的两个node，大的那个肯定就是要被recovery的其中之一，要记录。

另外一个，要遍历完整棵树，记录最后一个逆序的node。

简单而言，第一个逆序点要记录，最后一个逆序点要记录，最后swap一下。

因为Inorder用了递归来解决，所以为了能存储这两个逆序点，这里用了全局变量，用其他引用型遍历解决也可以。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    TreeNode pre;
    TreeNode first;
    TreeNode second;

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

        inorder(root.left);
        if (pre == null) {
            pre = root;  //pre指针初始
        } else {
            if (pre.val > root.val) {
                if(first == null) {
                    first = pre;//第一个逆序点
                }
                second = root;  //不断寻找最后一个逆序点
            }
            pre = root;  //pre指针每次后移一位
        }
        inorder(root.right);
    }  

    public void recoverTree(TreeNode root) {
        pre = null;
        first = null;
        second = null;
        inorder(root);
        if (first != null && second != null) {
            int tmp = first.val;
            first.val = second.val;
            second.val = tmp;
        }
    }
}