99. Recover Binary Search Tree

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void recoverTree(TreeNode* root) {
        TreeNode *pre=NULL,*cur=root;
        TreeNode *first=NULL,*second=NULL,*temp=NULL;
        while(cur!=NULL){
            if(cur->left==NULL){
                //cur->val;
                if(temp!=NULL&&temp->val>cur->val){
                    if(!first)first=temp;
                    second=cur;
                }
                temp=cur;
                cur=cur->right;
            }
            else {
                pre=cur->left;
                while(pre->right!=NULL&&pre->right!=cur)
                pre=pre->right;
                //exist threaded tree
                if(pre->right==cur){
                    //print cur->val;
                    if(temp!=NULL&&temp->val>cur->val){
                        if(!first)first=temp;
                        second=cur;
                    }
                    temp=cur;
                    pre->right=NULL;
                    cur=cur->right;
                }
                //creating backend poinner to the current point
                else{
                    pre->right=cur;
                    cur=cur->left;
                }
            }
        }
        swap(first->val,second->val);
    }
};