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;
    }
};