lintcode面试题7:根据前序遍历和中序遍历树构造二叉树

给出中序遍历:[1,2,3]和前序遍历:[2,1,3]. 返回如下的树:

2
/
1 3

补充知识:

前序排列:根节点,左子节点,右子节点
中序排列:左子节点,根节点,右子节点
这里写图片描述

解题思路:

1.根据前序排列第一个数字确定根节点,并在中序排列中确定根节点位置
2.根据根节点的位置确定中序排列中左子树、右子树,中序排列中根节点之前的数字为左子树,根节点之后的部分为右子树
3.根据中序排列中左子树和右子树的长度确定在前序排列中的左子树、右子树
将左子树构建为一个容器,右子树构建为一个容器。
4.递归构建左子树、右子树,直到叶子节点(判断条件为左右子树分别没有子树)

TreeNode * buildTree(vector<int> &preorder, vector<int> &inorder) {
        // write your code here
        TreeNode *root = NULL;
        vector<int> preorder_l,preorder_r,inorder_l,inorder_r;
        int i,root_index=0;

        if(preorder.empty()!=1 || inorder.empty()!=1) {
            root = new TreeNode(preorder[0]); //  在前序队列中找根节点

            //  在中序队列中找出根节点位置
            for(i=0; i<inorder.size(); i++) {
                if(preorder[0] == inorder[i])
                    break;
                root_index++;
            }

            //  左右子树的前序、中序队列
            for(i=0; i<root_index; i++) {
                preorder_l.push_back(preorder[i+1]);
                inorder_l.push_back(inorder[i]);
            }
            for(i=root_index+1; i<inorder.size(); i++) {
                preorder_r.push_back(preorder[i]);
                inorder_r.push_back(inorder[i]);
            }
			//递归构建左右子树
            root->left = buildTree(preorder_l, inorder_l);
            root->right = buildTree(preorder_r, inorder_r);
        }
        return root;
    
        
        
    }

类型 *p=new 类型[];//分配新的内存,指针指向数组第一个元素

  • 3
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值