7-70 树的遍历 (25 分)
给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数N(≤30),是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。
输出格式:
在一行中输出该树的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。
输入样例:
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
输出样例:
4 1 6 3 5 7 2
该题是已知后序和中序 求层序
先用两个数组pre和in数组将后续和中序存起来,然后再利用递归把大问题化成小问题,因为后序是LRD所以最后一个一定是根节点,在中序LDR里面找到这个根节点之后,左边的就是在根节点左子树上的结点,右边的就是在根节点右子树上的结点,再利用数组l,r把根节点的左右结点存起来
#include<iostream>
#include<map>
#include<queue>
using namespace std;
map<int,int>l,r;
int hou[1010];//后序遍历数组
int in[1010];//中序遍历数组
int n;
queue<int>q;
int build(int hl,int hr,int il,int ir)
{
if(hl>hr||il>ir) return 0;
int root=il;
//找到后序遍历根节点在中序遍历中的位置
while(root<=ir&&in[root]!=hou[hr]) root++;
int cnt=root-il;
l[hou[hr]]=build(hl,hl+cnt-1,il,root-1);
r[hou[hr]]=build(hl+cnt,hr-1,root+1,ir);
return hou[hr];
}
void print(int root)//层序遍历输出
{
q.push(root);
cout<<root;
int sum=1;
if(sum!=n) cout<<" ";
while(q.size())
{
root=q.front();
q.pop();
if(l[root])
{
cout<<l[root];
q.push(l[root]);
sum++;
if(sum!=n) cout<<" ";
}
if(r[root])
{
cout<<r[root];
sum++;
q.push(r[root]);
if(sum!=n) cout<<" ";
}
if(sum==n) break;
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>hou[i];
}
for(int i=1;i<=n;i++)
{
cin>>in[i];
}
int root=build(1,n,1,n);
print(root);
return 0;
}
下面是已知前序和中序求层序:
#include<iostream>
#include<map>
#include<queue>
using namespace std;
map<int,int>l,r;
int pre[1010];//前序遍历数组
int in[1010];//中序遍历数组
int n;
queue<int>q;
int build(int pl,int pr,int il,int ir)
{
if(pl>pr||il>ir)
{
return 0;
}
int root=il;
while(root<=ir&&in[root]!=pre[pl])//找到前序遍历根节点在中序遍历的位置
{
root++;
}
int cnt=root-il;
l[pre[pl]]=build(pl+1,pl+cnt,il,root-1);
r[pre[pl]]=build(pl+cnt+1,pr,root+1,ir);
return pre[pl];
}
void print(int root)//层序遍历输出
{
q.push(root);
cout<<root<<" ";
int sum=1;
while(q.size())
{
root=q.front();
q.pop();
if(l[root])
{
cout<<l[root]<<" ";
q.push(l[root]);
sum++;
}
if(r[root])
{
cout<<r[root]<<" ";
sum++;
q.push(r[root]);
}
if(sum==n) break;
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>pre[i];
}
for(int i=1;i<=n;i++)
{
cin>>in[i];
}
int root=build(1,n,1,n);
print(root);
return 0;
}
//7
//10 9 7 8 12 11 13
//7 9 8 10 11 12 13