leetcode Recover Binary Search Tree

Two elements of a binary search tree (BST) are swapped by mistake.

Recover the tree without changing its structure.

这道题目理解不是很深，我们知道对BST中序遍历可以得到一个从大到小的排列，只要有两个元素的顺序不对就是这两个了吧。

在中序遍历中，我们会先会递归处理左子树，之后处理当前节点，之后再递归处理右子树。用first和second记录两个错序的节点。

我们用pre表示当前节点root，在中序遍历中的前一个元素。如果pre->val>root->val,则顺序不对了吧。所以如果first为空，first=pre，用second=root（这种情况是为了处理first和second紧邻的情况）。

<span style="font-family: Arial, Helvetica, sans-serif; font-size: 12px;">/*struct TreeNode {</span>

 *     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,*first=NULL,*second=NULL;
        sub(root,pre,first,second);
        swap(first,second);
        
    }
    void sub(TreeNode *root,TreeNode *&pre,TreeNode *&first,TreeNode *&second){
        if(root==NULL) return;
        sub(root->left,pre,first,second);
        if(pre!=NULL&&pre->val>root->val){
            if(first==NULL){
                first=pre;
            }
            second=root;
        }
        pre=root;
        sub(root->right,pre,first,second);
    }
    void swap(TreeNode* node1,TreeNode* node2){
        int temp=node2->val;
        node2->val=node1->val;
        node1->val=temp;
    }
};