剑指offer面试题6——重建二叉树(递归)

时间:2020-12-18 17:29:41

题目1385:重建二叉树

时间限制:1 秒

内存限制:32 兆

特殊判题:

提交:4326

解决:1287

题目描述:

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

 

 

 

输入:

输入可能包含多个测试样例,对于每个测试案例,

输入的第一行为一个整数n(1<=n<=1000):代表二叉树的节点个数。

输入的第二行包括n个整数(其中每个元素a的范围为(1<=a<=1000)):代表二叉树的前序遍历序列。

输入的第三行包括n个整数(其中每个元素a的范围为(1<=a<=1000)):代表二叉树的中序遍历序列。

 

输出:

对应每个测试案例,输出一行:

如果题目中所给的前序和中序遍历序列能构成一棵二叉树,则输出n个整数,代表二叉树的后序遍历序列,每个元素后面都有空格。

如果题目中所给的前序和中序遍历序列不能构成一棵二叉树,则输出”No”。

 

样例输入:
8
1 2 4 7 3 5 6 8
4 7 2 1 5 3 8 6
8
1 2 4 7 3 5 6 8
4 1 2 7 5 3 8 6
样例输出:
7 4 2 5 8 6 3 1 
No
答疑:
解题遇到问题?分享解题心得?讨论本题请访问
这道题因为前序遍历第一个点就是根节点,中序遍历的中间某个点的左边就是左子树的,右边就是右子树的,这样算下来就可以用递归来做
#include<iostream>
#include<vector>
using namespace std;

struct node
{
	int val;
	node* left;
	node* right;
	node(int va):val(va),left(NULL),right(NULL){}
};

vector<int> vec1;
vector<int> vec2;

//根据前序遍历中序遍历来构建二叉树
node* product(int a1,int b1,int a2,int b2)
{
	node* root=new node(vec1[a1]);//当前根节点的创建
	if(a1==b1)
		return root;
	int i=0;
	for(i=a2;i<=b2;i++)
		if(vec2[i]==vec1[a1])
			break;
	if(i>a2)
		root->left=product(a1+1,a1+i-a2,a2,i-1);//左子树的创建
	if(i<b2)
		root->right=product(a1+i-a2+1,b1,i+1,b2);//右子树的创建
	return root;
}

//后序遍历二叉树
void last_out(node* root)
{
	if(root->left!=NULL)
		last_out(root->left);
	if(root->right!=NULL)
		last_out(root->right);
	cout<<root->val<<' ';
	return;
}

int main()
{
	int ary1[8]={1,2,4,7,3,5,6,8};
	int ary2[8]={4,7,2,1,5,3,8,6};

	for(int i=0;i<8;i++)
		vec1.push_back(ary1[i]);
	for(int i=0;i<8;i++)
		vec2.push_back(ary2[i]);
	node* root=product(0,7,0,7);
	last_out(root);
}