Leetcode 99. 恢复二叉搜索树 C++

Leetcode 99. 恢复二叉搜索树

题目

二叉搜索树中的两个节点被错误地交换。

请在不改变其结构的情况下，恢复这棵树。

测试样例

1.

输入: [1,3,null,null,2]

   1
  /
 3
  \
   2

输出: [3,1,null,null,2]

   3
  /
 1
  \
   2

2.

输入: [3,1,4,null,null,2]

  3
 / \
1   4
   /
  2

输出: [2,1,4,null,null,3]

  2
 / \
1   4
   /
  3

题解

对于二叉搜索树而言，中序遍历的结果应该是严格递增的。根据题意可以知道，有两个节点被交换了，也就是会有两个位置不会严格递增。第一个地方会发生一次递减，错误节点应该是pre；第二个地方会发生一次递减，而这次是root。当然，这两次可能发生在同一个位置。我们只要分别找到这两个位置，然后交换即可。详细过程见代码

代码

	TreeNode *pre=NULL;		//记录中序遍历时，前一次遍历的节点
    TreeNode *one=NULL;		//记录第一次发生错误的错误节点
    TreeNode *two=NULL;		//记录第二次发生错误的错误节点
    bool search(TreeNode *root){
        if(root == NULL)    return false;
        if(search(root->left))  return true;
        if(pre != NULL && pre->val>root->val){
            if(one==NULL){
                one = pre;
                two = root;
            }else{
                two=root;
                return true;
            } 
        } 
        pre = root;
        if(search(root->right)) return true;
        return false;
    }
    void recoverTree(TreeNode* root) {
        search(root);
        swap(one->val,two->val);
    }

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