题目叙述
给你二叉搜索树的根节点 root ,该树中的两个节点被错误地交换。请在不改变其结构的情况下,恢复这棵树。
进阶:使用 O(n) 空间复杂度的解法很容易实现。你能想出一个只使用常数空间的解决方案吗?
分析
方法1: O ( n ) O(n) O(n)的时间复杂度,使用中序遍历
使用中序遍历,先从前往后找第一个逆序的位置,再往后找第二个逆序的位置。然后进行交换。
先存储第一对降序的,后面还有第二对的话,把第二对的第二个数重新赋值修改一下就行了: 即以两个变量来存储一个逆序对,以ans1来存储前者,ans2来存储后者。对于两个逆序对,其实我们最终是要交换第一个逆序对的ans1和第二个逆序对的ans2。
1 5 3 4 2 先存5 3, 后面4 2 又降序了,说明5 2得交换
代码
class Solution {
public:
//从前往后的中序遍历
//从后往前的中序遍历
TreeNode* ans1;
TreeNode* ans2;
TreeNode* pre = new TreeNode(INT_MIN);
void inOrderFront(TreeNode* node) {
if(!node) return;
inOrderFront(node->left);
if(node->val<pre->val){
if(!ans1) ans1 = pre;//保证ans1只赋值一次
ans2 = node;
}
pre = node;
inOrderFront(node->right);
}
void recoverTree(TreeNode* node) {
inOrderFront(node);
//交换两个指针的内容
int tmp = ans1->val;
ans1->val = ans2->val;
ans2->val = tmp;
}
};