LeetCode 99. Recover Binary Search Tree

题意：找出“BST”中不符合规律的两个结点。

解题思路：

1. 对于BST，中序遍历就是一个递增序列，因此，只要依赖于中序遍历就能够找到两个有问题的结点，然后交换结点的值就行了。

/**
 * 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) {
        stack<TreeNode*> s;
        TreeNode* tmp=root;
        TreeNode* t1,*t2;  //指示wrong position的结点
        bool flag=false;   //判断是第一次还是第二次
        TreeNode* pre=NULL;
        
        while(!s.empty() || tmp!=NULL){
            if(tmp!=NULL){
                s.push(tmp);
                tmp=tmp->left;
            }
            else{
                while(!s.empty() && s.top()->right==NULL){   
                    // 发生错误
                    if(pre!=NULL && pre->val>s.top()->val){
                        cout<<"gg"<<endl;
                        if(flag==false){
                            t1=pre;
                            t2=s.top();
                            flag=true;
                        }
                        else{
                            t2=s.top();
                        }
                    }
                    pre=s.top();
                    //cout<<pre->val<<endl;
                    s.pop();
                }
                if(!s.empty()){
                    // 发生错误
                    if(pre!=NULL && pre->val>s.top()->val){
                        cout<<"gg"<<endl;
                        if(flag==false){
                            t1=pre;   
                            t2=s.top();
                            flag=true;
                        }
                        else{
                            t2=s.top();
                        }
                    }
                    pre=s.top();
                    //cout<<pre->val<<endl;
                    tmp=s.top()->right;
                    s.pop();
                }
            }
        }
        
        int data=t1->val;
        t1->val=t2->val;
        t2->val=data;
        
    }
};

• 在这里，我采用的是非递归中序遍历，使用pre记录前驱，tmp记录当前结点。   
• 其实，这个题就是在建立中序线索二叉树。  https://www.cnblogs.com/yy-1046741080/p/11511263.html
• 对于中序遍历，弹栈的时候才是中序序列，因此，pre的改变是发生在弹栈的时候，而且需要先进行比较。
• 在数据中，因为错误可能存在接连两个结点：所以在31行和48行必须处理t2；否则就会漏元素。