PAT-CCCC练习：L2-006.树的遍历

L2-006. 树的遍历

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数N（<=30），是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

72 3 1 5 7 6 41 2 3 4 5 6 7

输出样例：

4 1 6 3 5 7 2

记录于此，作为后序中序制作树的复习。

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>

using namespace std;

struct node
{
	int l,r;
}tree[35];
int back[35],mid[35];

int build(int midl,int midr,int backl,int backr)//中序左右，后序左右 
{
	if(midl>midr)
		return 0;//尽头处左右子树设为0 
	
	int root=backr;
	int ptr;
	for(ptr=midl;mid[ptr]!=back[backr];ptr++);
	
	int len=ptr-midl;
	
	tree[root].l=build(midl,ptr-1,backl,backl+len-1);
	tree[root].r=build(ptr+1,midr,backl+len,backr-1);
	
	return root;
}
void fgo(int root)//层序遍历 
{
	queue<node>q;
	vector<int>v;
	
	q.push(tree[root]);
	v.push_back(root); 
	while(!q.empty())
	{
		node t=q.front();
		if(t.l!=0)
		{
			q.push(tree[t.l]);
			v.push_back(t.l);
		}	
		if(t.r!=0)
		{
			q.push(tree[t.r]);
			v.push_back(t.r); 
		}
		q.pop();
	} 
	
	int temp=v.size();
	for(int i=0;i<temp;i++)
	{
		printf("%d%c",back[v[i]],i==temp-1?'\n':' ');
	}
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&back[i]);//后序遍历 
	for(int i=1;i<=n;i++)
		scanf("%d",&mid[i]);//中序遍历 
		
	build(1,n,1,n);
	
	fgo(n);
	
	return 0;
}