题目详情
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
示例 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);}
};