构造二叉树-前序遍历

1008. 前序遍历构造二叉搜索树

思路

  • 方法1

    • 二叉搜索树,BST,左子树小于根,右子树大于根,如此递归定义。

    • BST的中序遍历,即按递增排序的序列。

    • 如此,通过先序遍历和中序遍历,即可唯一确定一棵树。

  • 方法2

    • 先序遍历,先根,BST,左子树小于根,右子树大于根。
    • 递归构造
  • 方法3

    • 在方法2的基础上,我们直接用区间递归构造即可。

代码

  • 方法1

    /**
     * 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 {
    private:
        unordered_map<int,int> cnt;
    public:
        TreeNode* bstFromPreorder(vector<int>& preorder) {
            vector<int>inorder(preorder);
            sort(inorder.begin(),inorder.end());
            for(int i=0;i<inorder.size();i++){
                cnt[inorder[i]]=i;
            }
            return build(preorder, 0, preorder.size()-1, inorder, 0, inorder.size()-1);
        }
        TreeNode* build(vector<int>& preorder, int preStart, int preEnd, vector<int>& inorder, int inStart, int inEnd){
            if(preStart>preEnd){
                return nullptr;
            }
            int rootval = preorder[preStart];
            int index = cnt[rootval];
            int leftsize = index - inStart;
            TreeNode* root = new TreeNode(rootval);
            root->left = build(preorder, preStart+1, preStart+leftsize,
                                inorder, inStart, index-1);
            root->right = build(preorder, preStart+leftsize+1, preEnd,
                                inorder, index+1, inEnd);
            return root;
        }
    };
    
  • 方法2

    /**
     * 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:
        TreeNode* bstFromPreorder(vector<int>& preorder) {
            return build(preorder, 0, preorder.size()-1);
        }
    
        TreeNode* build(vector<int>& preorder, int left, int right){
            if(left>right)
                return nullptr;
            int rootval = preorder[left];
            TreeNode* root = new TreeNode(rootval);
            int l = left, r = right;//如果 l=left+1,对于只有两个数的区间,就会出问题
            while(l<r){
                int mid = (l+r)/2+1;//保证不会一直 mid=l,陷入死循环
                if(preorder[mid] < rootval){
                    // 中间值小于根,往右边
                    l = mid;
                }else{
                    // 中间值大于根,往左边
                    r = mid-1;//保证不会一直 mid=r,陷入死循环
                }
            }
            root->left = build(preorder, left+1, l);
            root->right = build(preorder, l+1, right);
            return root;
        }
    };
    
  • 方法3

    /**
     * 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 {
    private:
        int idx=0;//保证函数内部的修改有效。
    public:
        TreeNode* bstFromPreorder(vector<int>& preorder) {
            return build(preorder, -1, 1e8+5);
        }
    
        TreeNode* build(vector<int>& preorder, int lower, int higher){
            if(idx==preorder.size()){
                return nullptr;
            }
            int val = preorder[idx];
            if(val<lower||val>higher)
                return nullptr;
            idx++;
            TreeNode* root = new TreeNode(val);
            root->left = build(preorder, lower, val);
            root->right = build(preorder, val, higher);
            return root;
        }
    };
    
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值