题目描述
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/recover-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
二叉搜索树中序遍历是一个有序序列。交换其中两个节点相当于在有序序列中交换了两个数,我们只需找到这两个数,把它们交换回来即可。有两种情况:
1.相邻两个数字交换:[1,2,3,4,5,6]交换2,3变为[1,3,2,4,5,6],只有一组逆序对,中序遍历记录前驱,然后找到这两数交换即可;
2.不相邻两个数字交换:[1,2,3,4,5,6]交换2,6变为[1,6,3,4,5,2],此时有两组逆序对,6,3和5,2,所以我们只需要找到这两组逆序对,然后把第一组前一个数字和第二组后一个数字进行交换即可。
所以需要一个记录前驱的指针pre,当前驱大于当前结点,此时是一个逆序对,用两个指针first,second记录;如果有第二个逆序对,更新second为当前结点,最后交换first,second指向结点的数值即可。
代码(c++)
/**
* 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 inorder(TreeNode* &root,TreeNode* &pre,TreeNode* &first,TreeNode* &second){
if(root==NULL) return;
inorder(root->left,pre,first,second);
if(pre==NULL) pre=root;
else{
if(pre->val>root->val){
if(first==NULL) first=pre;
second=root;
}
pre=root;
}
inorder(root->right,pre,first,second);
}
void recoverTree(TreeNode* root) {
TreeNode* pre=NULL;
TreeNode* first=NULL;
TreeNode* second=NULL;
inorder(root,pre,first,second);
swap(first->val,second->val);
}
};