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;
}