2024每日刷题(140)
Leetcode—99. 恢复二叉搜索树
回溯法实现代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
void recoverTree(TreeNode* root) {
TreeNode* pred = nullptr;
TreeNode* first = nullptr;
TreeNode* last = nullptr;
function<void(TreeNode*)> dfs = [&](TreeNode* node) {
if(node == nullptr) {
return;
}
dfs(node->left);
if(pred && node->val < pred->val) {
last = node;
if(first == nullptr) {
first = pred;
} else {
return;
}
}
pred = node;
dfs(node->right);
};
dfs(root);
swap(first->val, last->val);
}
};
运行结果
栈实现代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
void recoverTree(TreeNode* root) {
stack<TreeNode*> st;
TreeNode* first = nullptr;
TreeNode* last = nullptr;
TreeNode* pred = nullptr;
while(!st.empty() || root != nullptr) {
while(root != nullptr) {
st.push(root);
root = root->left;
}
root = st.top(), st.pop();
if(pred && root->val < pred->val) {
last = root;
if(first == nullptr) {
first = pred;
} else {
break;
}
}
pred = root;
root = root->right;
}
swap(first->val, last->val);
}
};
运行结果
之后我会持续更新,如果喜欢我的文章,请记得一键三连哦,点赞关注收藏,你的每一个赞每一份关注每一次收藏都将是我前进路上的无限动力 !!!↖(▔▽▔)↗感谢支持!