描述
给定节点数为 n 的二叉树的前序遍历和中序遍历结果,请重建出该二叉树并返回它的头结点。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建出如下图所示。
提示:
1.vin.length == pre.length
2.pre 和 vin 均无重复元素
3.vin出现的元素均出现在 pre里
4.只需要返回根结点,系统会自动输出整颗树做答案对比
示例
输入:
[1,2,4,7,3,5,6,8],[4,7,2,1,5,3,8,6]
返回值:
{1,2,3,4,#,5,6,#,7,#,#,8}
思路
先理清用前序+中序得出二叉树的过程,再运用递归方法
找到每个子树的根节点,再分出左子树和右子树,然后运用递归建立head->left和head->right
参考图片
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pre int整型一维数组
* @param preLen int pre数组长度
* @param vin int整型一维数组
* @param vinLen int vin数组长度
* @return TreeNode类
*/
struct TreeNode* reConstructBinaryTree(int* pre, int preLen, int* vin, int vinLen ) {
// write code here
if(preLen==0)
{
return NULL;
}
//初始化一个节点
struct TreeNode*head=(struct TreeNode*)malloc(sizeof(struct TreeNode));
//传入的先序遍历顺序第一个即为当前树的根节点
head->val=*pre;
//在中序遍历中寻找当前节点所在位置
int k=0;
int length_left=0;
int length_right=0;
while(pre[0]!=vin[k])
{
k++;
}
//左子树长度即为k;
length_left=k;
//右子树长度
length_right=vinLen-k-1;
//递归处理左子树和右子树
head->left=reConstructBinaryTree(pre+1,length_left,vin,length_left);
head->right=reConstructBinaryTree(pre+1+k,length_right,vin+k+1,length_right);
return head;
}
注
前序遍历:就先遍历它的一个根结点然后遍历它的左子树,最后遍历右子树
中序遍历:先遍历左子树在遍历根节点,最后遍历右子树