No.92 - LeetCode99 - 判断BST中交换的两个点

最新推荐文章于 2024-11-16 17:20:24 发布

原创最新推荐文章于 2024-11-16 17:20:24 发布 · 99 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种使用栈和双端队列的数据结构优化方法，来找出并修正二叉搜索树中两个错误排序的节点。通过两次深度优先搜索，分别定位到中序遍历中违反规则的第一个和第二个节点，并交换其值，恢复树的正确性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

找中序遍历中违反规则的两个点：

因为不知道节点数量，用vector实现：32ms，用deque实现：24ms

其实可以进一步优化，节省空间开销，两个dfs函数；
第一次找前面错误的点，即中序从前面数第一个比后面值大的点；
第二次找后面错误的点，即中序从后面数第一个比前面值小的点；

两个dfs函数，分为Ldfs和Rdfs ；
Ldfs返回当前子树最右节点，当first修改后退出。
Rdfs返回当前子树最左节点，当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:
    void recoverTree(TreeNode* root) {
        stack<TreeNode*> s;
        deque<TreeNode*> dq;
        map<TreeNode*,int> mp;
        s.push(root);
        while(!s.empty()){
            TreeNode* now = s.top();
            s.pop();
            if(mp[now] == 1) dq.push_back(now);
            else{
                mp[now] = 1;
                if(now->right != NULL) s.push(now->right);
                s.push(now);
                if(now->left != NULL) s.push(now->left);
            }
        }
        TreeNode* first = dq.front(); dq.pop_front();
        while(!dq.empty()){
            if(first->val > dq.front()->val){ dq.push_front(first);break; }
            first = dq.front();
            dq.pop_front();
        }
        TreeNode* second = dq.back(); dq.pop_back();
        while(!dq.empty()){
            if(second->val < dq.back()->val) break;
            second = dq.back();
            dq.pop_back();
        }
        swap(first->val,second->val);
    }
};