Leetcode#99 Recover Binary Search Tree

时间:2023-03-09 03:18:22
Leetcode#99	Recover Binary Search Tree

原题地址

中序遍历二叉搜索树,正常情况下所有元素都应该按递增排列,如果有元素被交换,则会出现前面元素大于后面的情况,称作反序。由于交换了两个节点,所以通常会有两处反序,但如果是两个相邻节点发生了交换,则只会有一次反序。

注意,如果有两次反序,前面那次一定是较大数,后面那次一定是较小数

交换的时候注意只需要交换value就行了,别傻不啦叽的去交换节点。

代码:

 vector<pair<TreeNode *, TreeNode *> > nodes;
TreeNode *prev; void inorderTraverse(TreeNode *curr) {
if (!curr)
return;
inorderTraverse(curr->left);
if (prev && prev->val > curr->val)
nodes.push_back(pair<TreeNode *, TreeNode *>(prev, curr));
prev = curr;
inorderTraverse(curr->right);
} void recoverTree(TreeNode *root) {
inorderTraverse(root);
if (nodes.size() == ) {
int tmp = nodes[].first->val;
nodes[].first->val = nodes[].second->val;
nodes[].second->val = tmp;
}
if (nodes.size() == ) {
int tmp = nodes[].first->val;
nodes[].first->val = nodes[].second->val;
nodes[].second->val = tmp;
}
}