二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
示例 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) 空间复杂度的解法很容易实现。
你能想出一个只使用常数空间的解决方案吗?
解法一:按照中序遍历将root遍历一遍,并将每个位置的地址和该节点的值分别保存到数组中。然后将保存节点值的数组从小到大排序,按地址的顺序赋给每个节点即可。
/**
* 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 {
public:
void recoverTree(TreeNode* root) {
if(root == nullptr)
return;
vector<int> vec;
inorder(root,vec);
sort(vec.begin(),vec.end());
int k = 0;
recover(root,vec,k);
}
void inorder(TreeNode* root,vector<int> &vec){//vec要使用引用
if(root == nullptr)
return;
inorder(root->left,vec);
vec.push_back(root->val);
inorder(root->right,vec);
}
void recover(TreeNode* root,vector<int> vec,int &k){
if(root == nullptr)
return;
recover(root->left,vec,k);
root->val = vec[k++];
recover(root->right,vec,k);
}
};
解法二:参考链接https://blog.csdn.net/sinat_35261315/article/details/78885322
/**
* 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 {
private:
TreeNode *pre,*fir,*sec;
public:
void recoverTree(TreeNode* root) {
if(root == NULL)
return;
TreeNode minNode(INT_MIN);
pre = &minNode;
fir = NULL;
sec = NULL;
inorder(root);
swap(fir->val,sec->val);
}
void inorder(TreeNode* root){//vec要使用引用
if(root == NULL)
return;
inorder(root->left);
//当上一个节点的值大于当前节点值时,说明上一个节点是错误的,影响了递增
if(fir==NULL && pre->val>root->val)
fir = pre;
//当firstNode记录完后,当前节点小于前面的节点,说明当前节点是错误的
if(fir!=NULL && pre->val>root->val)//注意这个判断.都是大于等于时为错,因为小于才是对的
sec = root;
pre = root; //prevNode记录上一个节点
inorder(root->right);
}
};