二叉搜索树(BST)的两个元素被错误地交换。 恢复树,而不改变它的结构。

217 篇文章 0 订阅
174 篇文章 2 订阅

本题源自leetcode

-----------------------------------------------------------

思路1,中序遍历 找出左子树大于父节点的值,再找出右子树小于父节点的值,然后交换,用指针pre保存中序遍历的父节点

    TreeNode* a=NULL;
    TreeNode* b=NULL;
    TreeNode* pre=NULL;
public:
    void recoverTree(TreeNode *root) {
        if(root==NULL)
            return;
       dfs(root);
       if(a&&b)
           swap(a->val,b->val);
    }
    //中序遍历
    void dfs(TreeNode* root){
        if(root==NULL)
            return;
        dfs(root->left);
        if(pre&&pre->val>root->val){
            if(a==NULL)    //a 指向第一个错误的节点(左子树节点大于根节点)
                a=pre;
            b=root;        //b指向根节点比右子树大的节点 
        }
        pre=root;
        dfs(root->right);
    }

  思路2 :

用额外的空间o(n) ,中序遍历现在的二叉树,然后排序,再把值赋给二叉树。

    void recoverTree(TreeNode *root) {
        if(root==NULL)
            return;
        vector<TreeNode*> trees;
        vector<int> inorderVal;
        Inorder(root,trees,inorderVal);
        sort(inorderVal.begin(),inorderVal.end());
        for(int i=0;i<trees.size();i++){
            trees[i]->val=inorder[i];
        }
    }
    void Inorder(TreeNode* root,vector<TreeNode*>& trees,vector<int>& inorderVal){
        if(root==NULL)
            return;
        Inorder(root->left,trees,inorderVal);
        trees.push_back(root);
        inorderVal.push_back(root->val);
        Inorder(root->right);
    }


  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值