105. 从前序与中序遍历序列构造二叉树

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如,给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

思路:了解先根遍历和中根遍历的特点。每次在先根遍历数组中最左边的就是二叉树的根,然后去中根遍历数组中找到根,根左边的就是左子树,根右边的就是右子树。对于左子树和右子树采取同样的求法:去先根遍历数组中找根,利用中根遍历数组划分左右子树,递归求解。

class Solution {
public:
    TreeNode* dfs(vector<int>& preorder, int l1, int r1, vector<int>& inorder, int l2, int r2){
        if(l1>r1) return nullptr;
        TreeNode *root=new TreeNode(preorder[l1]);
        for(int i=l2; i<=r2; ++i){
            if(inorder[i]==preorder[l1]){
                root->left=dfs(preorder, l1+1, l1+i-l2, inorder, l2, i-1);
                root->right=dfs(preorder, l1+i-l2+1, r1, inorder, i+1, r2);
                break;
            }
        }
        return root;
    }

    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        return dfs(preorder, 0, preorder.size()-1, inorder, 0, preorder.size()-1);
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值