07 重建二叉树

题目描述

根据二叉树的前序遍历和中序遍历的结果,重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

解题思路

前序遍历的第一个值为根节点的值,使用这个值将中序遍历结果分成两部分,左部分为树的左子树中序遍历结果,右部分为树的右子树中序遍历的结果。

代码

流程

——新建一个用来递归的binarytree函数,传入的参数包括:前序遍历数组,起点,终点(该值取不到),中序遍历数组,起点,终点
return BinaryTree(pre, 0, pre.size(), vin, 0, vin.size());
————前序遍历数组的第一个位置的数是根节点;
————遍历中序数组,找到与根节点相等的值的index,记为p_vin;
————使用p_vin将二叉树分成左右两个子树
——————左子树: BinaryTree(pre, pre_begin+1, pre_begin+1+(p_vin-vin_begin), vin, vin_begin, p_vin);
——————右子树:BinaryTree(pre, pre_begin+1+(p_vin-vin_begin), pre_end, vin, p_vin+1, vin_end);

代码片

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        return BinaryTree(pre, 0, pre.size(), vin, 0, vin.size());
    }
    TreeNode* BinaryTree(vector<int> pre, int pre_begin, int pre_end, vector<int> vin, int vin_begin, int vin_end)//注意函数声明是,各个形参要带着函数类型
    {
        if((pre_begin>=pre_end)||(vin_begin>=vin_end))
        {
            return NULL;//这里必须使用NULL
        }
        
        int p_vin = vin_begin;
        for(; p_vin < vin_end; p_vin++)
        {
        	//在中序遍历中寻找根节点
            if(vin[p_vin] == pre[pre_begin])//==判断是否相等是两个等号切记!!!
                break;
        }
		
		//重建二叉树的根节点
        //TreeNode* root = pre[pre_begin];//这一句整个就错了,正确写法见下面
        TreeNode* root = new TreeNode(pre[pre_begin]);
        //在p_vin的位置处将中序遍历分割成左右子树
        root -> left = BinaryTree(pre, pre_begin+1, pre_begin+1+(p_vin-vin_begin), vin, vin_begin, p_vin);//参数“int pre_end”是取不到的,所以pre_begin+1+(p_vin-vin_begin)并不会被取到
        root -> right = BinaryTree(pre, pre_begin+1+(p_vin-vin_begin), pre_end, vin, p_vin+1, vin_end); //至于为什么左子树和右子树都不取到p_vin?p_vin是根节点啊,已经成了root了。
        return root;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值