LeetCode 99. Recover Binary Search Tree (BST重建)

最新推荐文章于 2021-06-13 13:54:56 发布

HuangHongkai_

最新推荐文章于 2021-06-13 13:54:56 发布

阅读量196

点赞数

分类专栏：树 LeetCode

本文链接：https://blog.youkuaiyun.com/qq_26973089/article/details/83616862

版权

LeetCode 同时被 2 个专栏收录

18 篇文章

订阅专栏

树

5 篇文章

订阅专栏

本文介绍了一种简单有效的算法，用于解决二叉搜索树中因两个元素错误交换而引发的问题。通过两次中序遍历，第一次获取原始序列并排序，第二次则根据排序后的序列重构树结构，确保了树的正确性和效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

Recover the tree without changing its structure.

Example 1:

Input: [1,3,null,null,2]

   1
  /
 3
  \
   2

Output: [3,1,null,null,2]

   3
  /
 1
  \
   2

Example 2:

Input: [3,1,4,null,null,2]

  3
 / \
1   4
   /
  2

Output: [2,1,4,null,null,3]

  2
 / \
1   4
   /
  3

Follow up:

A solution using O(n) space is pretty straight forward.
Could you devise a constant space solution?

解法
本来想了一些稍复杂的算法，后来发现完全不需要，利用二叉搜索树的排序性质，中序遍历即可得到排序的序列。先一个遍历获取到序列，然后对序列排序，最后再来一个中序遍历重构二叉树（只需要该节点值即可）

/**
 * 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 {
    int i;
    void InOrderVis(TreeNode* root, vector<int>& node) {
        if(root) {
            InOrderVis(root->left, node);
            node.push_back(root->val);
            InOrderVis(root->right, node);
        }
    }
    
    void InOrderVisBuild(TreeNode* root, vector<int>& node) {
        if(root) {
            InOrderVisBuild(root->left, node);
            root->val = node[i++];
            InOrderVisBuild(root->right, node);
        }
    }
public:
    void recoverTree(TreeNode* root) {
        vector<int> node;
        InOrderVis(root, node);
        sort(node.begin(), node.end());
        i=0;
        InOrderVisBuild(root, node);
    }
};