leetcode 99: Recover Binary Search Tree

最新推荐文章于 2020-10-22 11:11:19 发布

转载最新推荐文章于 2020-10-22 11:11:19 发布 · 280 阅读

leetcode 专栏收录该内容

201 篇文章

订阅专栏

本文介绍了一种通过中序遍历来检测并修正二叉搜索树中两个被错误交换节点的方法。具体实现中，利用递归进行中序遍历，并在遍历过程中比较当前节点值与前一个节点值的关系来确定错误节点。

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

Learned from http://www.cnblogs.com/tgkx1054/archive/2013/05/24/3096830.html. Do the inorder traversal of the tree, and we will always get root->val less than pre->val. If root->val > pre->val, mistake is found.

/**
 * 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* pre;
    TreeNode* p,*q;
    void recoverTree(TreeNode* root) {
        p=q=pre=NULL;
        inorder(root);
        swap(p->val,q->val);
    }
    void inorder(TreeNode* root)
    {
        if(!root)
            return;
        if(root->left)
            inorder(root->left);
        if(pre&&pre->val>root->val)
        {
            if(!p)
                p=pre;
            q=root;
        }
        pre=root;
        if(root->right)
            inorder(root->right);
    }
};