当前位置: 代码迷 >> 综合 >> 99. 恢复二叉搜索树(C++)---中序遍历解法
  详细解决方案

99. 恢复二叉搜索树(C++)---中序遍历解法

热度:94   发布时间:2024-02-08 06:06:09.0

题目详情

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

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

示例 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

进阶:

  • 使用 O(n) 空间复杂度的解法很容易实现。
  • 你能想出一个只使用常数空间的解决方案吗?

 

——题目难度:困难






-下面代码

class Solution {
public:void recoverTree(TreeNode* root) {vector<int> result;inOrder(root, result); //中序遍历sort(result.begin(), result.end());recover(root, result);}void inOrder(TreeNode* root, vector<int>& result) {if (root == NULL) return ;inOrder(root->left, result);result.push_back(root->val);inOrder(root->right, result);}void recover(TreeNode* root, vector<int>& result) { //中序遍历if (root == NULL) return ;recover(root->left, result);if (root->val == result[0]) {result.erase(result.begin());} else {root->val = result[0];result.erase(result.begin());}recover(root->right, result);}
};

 

  相关解决方案