【leetcode】99. Recover Binary Search Tree【java】简单又容易理解的中序遍历的方法

Two elements of a binary search tree (BST) are swapped by mistake.

Recover the tree without changing its structure.

Note:

A solution using O(n) space is pretty straight forward. Could you devise a constant space solution?

这个问题似乎很难，但它仅仅是一个简单的中序遍历！

让我们从中序遍历开始入手

private void traverse (TreeNode root) {
   if (root == null)
      return;
   traverse(root.left);
   // Do some business
   traverse(root.right);
}

什么是我们在这里做的business呢？
我们需要找到第一个和第二个不在正确的顺序上的元素

我们如何发现这两个元素呢？例如，我们有如下树，中序遍历并打印为：

6，3，4，5，2

我们比较每个节点与它的下一个，我们可以发现6是第一个需要交换的元素，因为6> 3；2是第二个需要交换的元素，因为2 <5；的第二个元素的第一个元素。

实际上，我们是在比较的是当前节点及其之前的“中序遍历”的节点。

让我们定义三个变量，firstElement，secondElement和prevElement。现在我们只在“do some business”逻辑寻找两个元素。请参见下面的代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    TreeNode firstElement = null;
    TreeNode secondElement = null;
    TreeNode preElement = new TreeNode(Integer.MIN_VALUE);
    public void recoverTree(TreeNode root) {
        traverse(root);
        int tmp = firstElement.val;
        firstElement.val = secondElement.val;
        secondElement.val = tmp;
    }
    public void traverse(TreeNode root){
        if (root == null){
            return;
        }
        traverse(root.left);
        if (firstElement == null && preElement.val >= root.val){
            firstElement = preElement;
        }
        if (firstElement != null && preElement.val >= root.val){
            secondElement = root;
        }
        preElement = root;
        traverse(root.right);
    }
}