PAT甲级1020 Tree Traversals

PAT甲级1020 Tree Traversals

题目大意

本题为简单的树形数据结构问题，根据后续遍历和中序遍历，求出层次遍历。

解题思路

后续遍历的顺序，最后一个为根节点，倒数第二个为根节点的右子树的根节点，而第n-r-1位对应根节点的左子树的根节点(n为节点总数,r为右子树所有节点的总数)，因此我们可以根据上述规律进行递归，构造出一颗树，然后再进行层次遍历。

层次遍历思路

用队列存储节点。刚开始让根节点入队，然后每次将队列进行出队操作时，把出队的节点的左右孩子节点加入到队列里，队列空则停止。

下附代码

#include<bits/stdc++.h>
#define N 100
using namespace std;
//树的节点结构体
struct node{
	node *r=NULL;
	node *l=NULL;
	int id;
};
int n,a[N],b[N];
queue<node *>c;
//构造树，pos为当前对应的后续遍历的节点位置，即当前子树的根节点
//l为该子树在中序遍历中的左边界，r为右边界
node * fun(int pos,int l,int r){
	node* node1=new node;
	node1->id=a[pos];
	if(l==r)
		return node1;
	if(l>r||l<0||r>n)
		return NULL;
	for(int i=l;i<=r;i++)
		if(a[pos]==b[i]){
			node1->r=fun(pos-1,i+1,r);
			node1->l=fun(pos-(r-i)-1,l,i-1);
			break;
		}
	return node1;
}

//层次遍历
void fun2(node *root){
	c.push(root);
	while(!c.empty()){
		if(c.front()->l!=NULL)
			c.push(c.front()->l);
		if(c.front()->r!=NULL)
			c.push(c.front()->r);
		cout<<c.front()->id;
		if(c.size()!=1)
			cout<<" ";
		else
			cout<<endl;
		c.pop();
	}
	
	
}
int main(){
	node* root;
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<=n;i++)
		cin>>b[i];
	root=fun(n,1,n);
	fun2(root);
	
	return 0;
}

希望大家都可以顺利做出本题