Recover Binary Search Tree

http://blog.csdn.net/havenoidea/article/details/12869021

题意二叉查找树不合法，有两个节点的值被交换了，找出这两个节点并且不改变树的结构，使得二叉查找树合法，常数空间限制。

解题思路：中序遍历查找树会得到一个递增的数列。在中序遍历过程中，找出不符合递增条件的节点，然后交换它们的值。

class Solution {
public:
    Solution(): pre(NULL), s1(NULL), s2(NULL){}
    void recoverTree(TreeNode *root) {
        if(!root)
          return;
        pre = NULL; s1 = NULL; s2 = NULL;
        findRecoverNode(root);
        if(s1 && s2) {
            swap(s1->val, s2->val);
        }
    }

private:
    TreeNode *pre;
    TreeNode *s1;
    TreeNode *s2;
    void findRecoverNode(TreeNode *root) {
        if(!root)
          return;
        findRecoverNode(root->left);
        if(pre && pre->val > root-> val) {
            if(!s1) {
                s1 = pre;
                s2 = root;
            } else {
                s2 = root;
            }
        }
        pre = root;
        findRecoverNode(root->right);
    }
};