Recover Binary Search

原文地址：http://blog.csdn.net/worldwindjp/article/details/21694179

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?
http://oj.leetcode.com/problems/recover-binary-search-tree/

二叉排序树中有两个节点被交换了，要求把树恢复成二叉排序树。
最简单的办法，中序遍历二叉树生成序列，然后对序列中排序错误的进行调整。最后再进行一次赋值操作。
但这种方法要求n的空间复杂度，题目中要求空间复杂度为常数，所以需要换一种方法。
递归中序遍历二叉树，设置一个pre指针，记录当前节点中序遍历时的前节点，如果当前节点大于pre节点的值，说明需要调整次序。
有一个技巧是如果遍历整个序列过程中只出现了一次次序错误，说明就是这两个相邻节点需要被交换。如果出现了两次次序错误，那就需要交换这两个节点。

AC代码：

[java]  view plain copy

1. public class Solution {  
2.     TreeNode mistake1, mistake2;;  
3.     TreeNode pre;  
4.       
5.     void recursive_traversal(TreeNode root) {  
6.         if(root==null) {  
7.             return;  
8.         }  
9.         if(root.left!=null) {  
10.             recursive_traversal(root.left);  
11.         }  
12.         if(pre!=null&&root.val<pre.val) {  
13.             if(mistake1==null) {  
14.                 mistake1 = pre;  
15.                 mistake2 = root;  
16.             } else {  
17.                 mistake2 = root;  
18.             }  
19.         }  
20.         pre = root;  
21.         if(root.right!=null) {  
22.             recursive_traversal(root.right);  
23.         }  
24.     }  
25.     public void recoverTree(TreeNode root) {  
26.         //pre必须设为null，通过遍历的时候设进去。因为是中序遍历，所以pre应该是深层叶子左子树的父节点。  
27.         recursive_traversal(root);  
28.         if(mistake1!=null&&mistake2!=null) {  
29.             int tmp = mistake1.val;  
30.             mistake1.val = mistake2.val;  
31.             mistake2.val = tmp;  
32.         }  
33.     }  
34. }