leetcode_105题——Construct Binary Tree from Preorder and Inorder Traversal（树，递归）

Construct Binary Tree from Preorder and Inorder Traversal

Total Accepted: 32279 Total Submissions: 122094My Submissions

Question Solution

Given preorder and inorder traversal of a tree, construct the binary tree.

Note:
You may assume that duplicates do not exist in the tree.

Hide Tags

Tree Array Depth-first Search

Have you met this question in a real interview?

Yes

No

Discuss

这道题给了你二叉树前序遍历和中序遍历的结果（以vector的形式给的，其中没有那些不存在的点），所以在前序遍历中第一个肯定是根节点

这道题采用递归的方法来做，因为在前序遍历中的第一个点，在中序遍历中，它的左边为其左子树，右边为其右子树，这样依次向下递归就可以构造出了二叉树。

下面是AC的解法

#include<iostream>
#include <vector>
using namespace std;

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

/*找到a在vec中的位置*/
int laocation_vector(vector<int>& vec,int a,int i,int j)
{
	for(int k=i;k<=j;++k)
		if(vec[k]==a)
			return k;
	return -1;
}

/*i,j是范围，k是开始计算第几个*/
void pre_construct(vector<int>& preorder,vector<int>& inorder,TreeNode** root,int i,int j,int& k)
{
	if(k>=preorder.size())
	{
		*root=NULL;
		return;
	}
	int loacte_in=laocation_vector(inorder,preorder[k],i,j);
	if(loacte_in<i||loacte_in>j)
	{
		*root=NULL;
		return;
	}
	*root=(TreeNode*)malloc(sizeof(TreeNode));
	(*root)->val=preorder[k];
	k++;
	pre_construct(preorder,inorder,&(*root)->left,i,loacte_in-1,k);
	pre_construct(preorder,inorder,&(*root)->right,loacte_in+1,j,k);
	return;
}
/*这个函数来主要调用*/
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder)
{
	TreeNode* root=NULL;
	if(preorder.size()==NULL)
		return root;
	int k=0;
	pre_construct(preorder,inorder,&root,0,(preorder.size()-1),k);
	return root;
}
int main()
{
	vector<int> vec_pre;
	vector<int> vec_in;
	vec_pre.push_back(1),vec_pre.push_back(2);
	vec_in.push_back(2),vec_in.push_back(1);
	TreeNode* root;
	root=buildTree(vec_pre,vec_in);
}