恢复二叉搜索树

题目描述

二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下，恢复这棵树。


来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/recover-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

二叉搜索树中序遍历是一个有序序列。交换其中两个节点相当于在有序序列中交换了两个数，我们只需找到这两个数，把它们交换回来即可。有两种情况：
1.相邻两个数字交换：[1,2,3,4,5,6]交换2,3变为[1,3,2,4,5,6]，只有一组逆序对，中序遍历记录前驱，然后找到这两数交换即可；
2.不相邻两个数字交换：[1,2,3,4,5,6]交换2,6变为[1,6,3,4,5,2]，此时有两组逆序对，6,3和5,2，所以我们只需要找到这两组逆序对，然后把第一组前一个数字和第二组后一个数字进行交换即可。
所以需要一个记录前驱的指针pre，当前驱大于当前结点，此时是一个逆序对，用两个指针first，second记录；如果有第二个逆序对，更新second为当前结点，最后交换first，second指向结点的数值即可。

代码（c++）

/**
 * 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 inorder(TreeNode* &root,TreeNode* &pre,TreeNode* &first,TreeNode* &second){
        if(root==NULL) return;
        inorder(root->left,pre,first,second);
        if(pre==NULL) pre=root;
        else{
            if(pre->val>root->val){
                if(first==NULL) first=pre;
                second=root;
            }
            pre=root;
        }
        inorder(root->right,pre,first,second);
    }
    void recoverTree(TreeNode* root) {
        TreeNode* pre=NULL;
        TreeNode* first=NULL;
        TreeNode* second=NULL;
        inorder(root,pre,first,second);
        swap(first->val,second->val);
    }
};