7-6 根据后序和中序遍历输出先序遍历 (10 分)
本题要求根据给定的一棵二叉树的后序遍历和中序遍历结果,输出该树的先序遍历结果。
输入格式:第一行给出正整数N(≤30),是树中结点的个数。随后两行,每行给出N个整数,分别对应后序遍历和中序遍历结果,数字间以空格分隔。题目保证输入正确对应一棵二叉树。
输出格式:在一行中输出Preorder: 以及该树的先序遍历结果。数字间有1个空格,行末不得有多余空格。
输入样例:7 2 3 1 5 7 6 4 1 2 3 4 5 6 7
输出样例:
Preorder: 4 1 3 2 6 5 7
思路:
1、首先根据当前子树的后序遍历结果得出当前子树的根结点——即该遍历结果的最后一个元素。
2、在当前子树的中序遍历结果中得出其左子树与右子树的中序遍历结果,这里用两个变量来记录首尾区间
3、在当前子树的后序遍历结果中得出其左子树与右子树的遍历结果——这里用两个变量来记录首尾区间
4、递归进入左右子树
5、重复上述步骤
6、按照先序遍历的规则打印
核心思想:
根据一个树的后序遍历得出树的根结点,再利用该树的中序遍历结果得出左右子树的后序遍历结果,然后递归下去。
就是一个不断递归的过程
#include<iostream>
using namespace std;
int data1[107],data2[107];
struct TNode
{
int val=-1;
TNode *ltree=NULL;
TNode *rtree=NULL;
};
void solve(TNode *now,int low1,int hei1,int low2,int hei2)
{//用两对变量分别表示当前子树的后序遍历区间和中序遍历区间
if(low2==hei2)return;
now->ltree=new TNode;
now->rtree=new TNode;
now->val=data1[hei1-1];//根结点是后序遍历的最后一个
int i;
for(i=low2;i<hei2;i++)
{
if(data2[i]==data1[hei1-1])break;//找到根结点的值
}
//递归传参
solve(now->ltree,low1,low1-low2+i,low2,i);//进入左子树
solve(now->rtree,low1-low2+i,hei1-1,i+1,hei2);//进入右子树
}
void Tprint(TNode *now)//按照先序遍历的顺序打印
{
cout<<" "<<now->val;
if(now->ltree->val!=-1)Tprint(now->ltree);
if(now->rtree->val!=-1)Tprint(now->rtree);
return;
}
int main()
{
TNode *root=new TNode;
int n;
cin>>n;
for(int i=0;i<n;i++)cin>>data1[i];//后序
for(int i=0;i<n;i++)cin>>data2[i];//中序
solve(root,0,n,0,n);
cout<<"Preorder:";
Tprint(root);
return 0;
}
这里有二叉树的笔记 二叉树