Recover Binary Search Tree leetcode 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?

 

题解：

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

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

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

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

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

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

 

代码如下：

 

 1  public  class Solution {
 2     TreeNode pre;
 3     TreeNode first;
 4     TreeNode second;
 5       
 6      public  void inorder(TreeNode root){  
 7          if(root ==  null)  
 8              return;  
 9 
10         inorder(root.left);  
11          if(pre ==  null){  
12             pre = root;   // pre指针初始
13          } else{  
14              if(pre.val > root.val){  
15                  if(first ==  null){  
16                     first = pre; // 第一个逆序点
17                  }  
18                 second = root;   // 不断寻找最后一个逆序点
19              }  
20             pre = root;   // pre指针每次后移一位
21          }  
22         inorder(root.right);  
23     }  
24       
25      public  void recoverTree(TreeNode root) {  
26         pre =  null;  
27         first =  null;  
28         second =  null;  
29         inorder(root);  
30          if(first!= null && second!= null){   
31              int tmp = first.val;  
32             first.val = second.val;  
33             second.val = tmp;  
34         }  
35     }