Recover Binary Search Tree

class Solution {

public:

TreeNode *s1,*s2,*pre;

void recoverTree(TreeNode* root) {

if(!root)return ;

s1=s2=pre=NULL;

find(root);

swap(s1->val,s2->val);

}

void find(TreeNode* root)

{

if(root==NULL) return;

find( root->left);

if(pre&&pre->val>root->val)

{

if(s1==NULL) s1=pre,s2=root;

else s2=root;

}

pre=root;

find(root->right);

}

};

中序遍历时pre 始终指向root的前一个。