L2-006. 树的遍历
给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数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
后序遍历的最右边的为根节点,找到根节点在中序遍历的位置。
该位置的后边的点为左子树,后边为右之树。
推荐(二叉树的重建讲解):http://blog.csdn.net/hacker_zhidian/article/details/60770262
数组存树1:
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<string>
#include<sstream>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<vector>
using namespace std;
#define inf 0x3f3f3f3f
#define LL long long
int ed[50],in[50];
int L[50],R[50];
int dfs(int l1,int r1,int l2,int r2)
{
if(l1>r1)return 0;
int root=ed[r1];
int p=l2;
while(root!=in[p])p++;
int cnt=p-l2;
L[root]=dfs(l1,l1+cnt-1,l2,p-1);
R[root]=dfs(l1+cnt,r1-1,p+1,r2);
return root;
}
int main()
{
int n,i,j;
memset(L,0,sizeof L);
memset(R,0,sizeof R);
scanf("%d",&n);
for(i=1;i<=n;i++)
scanf("%d",&ed[i]);
for(i=1;i<=n;i++)
scanf("%d",&in[i]);
dfs(1,n,1,n);
//printf("%d %d\n",L[4],R[4]);
queue<int> q;
q.push(ed[n]);
while(!q.empty())
{
int u=q.front();
q.pop();
if(u!=ed[n])printf(" ");
printf("%d",u);
if(L[u]!=0)
q.push(L[u]);
if(R[u]!=0)
q.push(R[u]);
}
}
数组存树2:
#include<stdio.h>
#include<string.h>
#include<string>
#include<math.h>
#include<queue>
#include<iostream>
#include<algorithm>
using namespace std;
int mid[1005],beh[1005],tree[10000],tr[10000];
void build(int n,int la,int ra,int lb,int rb)
{
if(ra<la||rb<lb)
return;//防止出界
if(la==ra)
{
tree[n]=mid[la];
return;
}
for(int i=la;i<=ra;i++)
{
if(mid[i]==beh[rb])
{
tree[n]=beh[rb];
build(2*n,la,i-1,lb,lb+i-1-la);
build(2*n+1,i+1,ra,lb+i-la,rb-1);
break;
}
}
}
void bfs(int s)//层次遍历
{
int ans=1,y;
queue<int>Q;
Q.push(s);
while(!Q.empty())
{
y=Q.front();
Q.pop();
if(tree[y]==0) continue;
tr[ans++]=tree[y];
Q.push(2*y);
Q.push(2*y+1);
}
}
int main()
{
int n,i,j;
scanf("%d",&n);
for(i=1;i<=n;i++)
scanf("%d",&beh[i]);
for(i=1;i<=n;i++)
scanf("%d",&mid[i]);
build(1,1,n,1,n);
bfs(1);
for(i=1;i<n;i++)
printf("%d ",tr[i]);
printf("%d\n",tr[n]);
}
结构体存树:
#include<stdio.h>
#include<string.h>
#include<string>
#include<math.h>
#include<queue>
#include<iostream>
#include<algorithm>
using namespace std;
int mid[1005],beh[1005],tr[10000];
struct Node
{
int l,r;
}tree[1005];
int build(int la,int ra,int lb,int rb)
{
if(ra<la||rb<lb)
return 0;//防止出界
int root=beh[rb],p=la,pp;
while(mid[p]!=root)
p++;
pp=p-la;
tree[root].l=build(la,p-1,lb,lb+pp-1);
tree[root].r=build(p+1,ra,lb+pp,rb-1);
return root;
}
void bfs(int s)//层次遍历
{
int ans=1,y;
queue<int>Q;
Q.push(s);
while(!Q.empty())
{
y=Q.front();
Q.pop();
if(y==0) continue;
tr[ans++]=y;
Q.push(tree[y].l);
Q.push(tree[y].r);
}
}
int main()
{
int n,i,j;
scanf("%d",&n);
for(i=1;i<=n;i++)
scanf("%d",&beh[i]);
for(i=1;i<=n;i++)
scanf("%d",&mid[i]);
build(1,n,1,n);
bfs(beh[n]);
for(i=1;i<n;i++)
printf("%d ",tr[i]);
printf("%d\n",tr[n]);
}