恢复二叉搜索树
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
示例 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
使用语言:C++
思路:二叉搜索树的性质为采用中序遍历得到的序列为从小到大,
于是采用中序遍历,寻找位置不对的first和second并将其进行交换
/**
* 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:
TreeNode* first=NULL;
TreeNode* second=NULL;
TreeNode* pre =NULL;
void inorder(TreeNode* root){
if(!root) return;
inorder(root->left);
if(pre&&pre->val>root->val){
if(!first){
first=pre;
second=root;
}
else{
second=root;
}
}
pre=root;
inorder(root->right);
}
void recoverTree(TreeNode* root) {
inorder(root);
int temp=first->val;
first->val=second->val;
second->val=temp;
}
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/recover-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。