剑指offer:重建二叉树的C++实现

最新推荐文章于 2021-04-05 13:34:27 发布

半梦半醒间幸运_

最新推荐文章于 2021-04-05 13:34:27 发布

阅读量132

点赞数

分类专栏： c++算法代码

本文链接：https://blog.csdn.net/weixin_43086349/article/details/107830683

版权

c++算法代码专栏收录该内容

114 篇文章 1 订阅

订阅专栏

在这里插入图片描述

struct TreeNode
{
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int x):val(x),left(NULL),right(NULL){}
};

TreeNode* built(vector<int>& preorder,vector<int>& inorder)
{
	if(preorder.size()==0 || inorder.size()==0 || preorder.size()!=inorder.size()) return NULL;
	return build(preorder,0,preorder.size()-1,inorder,0,inorder.size()-1);
}

TreeNode* build(vector<int>& preorder,int left1,int right1,vector<int>& inorder,int left2,int right2)
{
	//建立根节点
	TreeNode* root=new TreeNode(preorder[left1]);
	int i=left2;
	while(preorder[left1]!=inorder[i]) i++;
	int left_len=i-left2;//左子树长度
	int right_len=right2-i;//右子树长度
	if(left_len>0) root->left=build(preorder,left1+1,left1+left_len,inorder,left2,i-1);
	if(right_len>0) root->right=build(preorder,left1+left_len+1,right1,inorder,i+1,right2);
	return root;
}