您的位置:首页 > 职场人生

【剑指Offer面试编程题】题目1385:重建二叉树--九度OJ

2014-12-18 20:49 417 查看
题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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


【解题思路】结合给出的图片和第一个样例输入来说,从前序输入的第一个元素1在中序输入中的位置可以确定中序输入中1之前的2 4 7三个元素为左子树,同理确定1之后的5 3 8 6四个元素为右子树。

根据以上相同的思路,前序输入中左字数2 4 7的第一个元素2确定为左子树的根元素,在后序输入中确定2之前的元素为左子树的左支,因为2之后没有元素,所以以2为根的树没有右支。同理递归的处理其他元素。最后形成一个完成的树,然后对树递归的进行后序遍历即可得到结果。

题目利用了前序遍历和中序遍历的特点,另外一种情况是给出后序和中序的结果也能确定一个树。而对树的处理,多为递归操作。

AC code:

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

vector<int> veci;
bool flag=true;

void back(const vector<int>vec1,const vector<int> vec2,const int&l1,const int &r1,const int &l2,const int &r2)
{
int tt=vec1[l1];
int i;
for(i=l2;i<=r2;++i)
if(tt==vec2[i])
break;
if(i>r2)
{
flag=false;
return;
}
int len=i-l2;
if(i>l2)
back(vec1,vec2,l1+1,l1+len,l2,i-1);
if(i<r2)
back(vec1,vec2,l1+len+1,r1,i+1,r2);
veci.push_back(tt);
}

int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
vector<int> vec1(n);
vector<int> vec2(n);
for(int i=0;i<n;++i)
scanf("%d",&vec1[i]);
for(int i=0;i<n;++i)
scanf("%d",&vec2[i]);
veci.clear();
flag=true;
back(vec1,vec2,0,n-1,0,n-1);
if(flag)
for(int i=0;i<veci.size();++i)
printf("%d ",veci[i]);
else
printf("No");
printf("\n");
}
return 0;
}
/**************************************************************
Problem: 1385
User: huo_yao
Language: C++
Result: Accepted
Time:10 ms
Memory:1524 kb
****************************************************************/题目链接:http://ac.jobdu.com/problem.php?pid=1385
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  zjof 面试 二叉树 递归
相关文章推荐