PAT.A1020 Recorridos de árboles

Volver a contenidos

Titulo

Dada la secuencia del recorrido de orden posterior y el recorrido de orden medio de un árbol, encuentre la secuencia de recorrido jerárquico.

Muestra (se puede copiar)

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

//output
4 1 6 3 5 7 2

Puntos a tener en cuenta

1. El orden de varios recorridos. Recorrido de primer orden: centro-izquierda-derecha. Recorrido de orden medio: izquierda-centro-derecha. Recorrido posterior al pedido: izquierda-derecha-centro.
2. Se recomienda que el lector simule manualmente la función de creación para construir el árbol dado en el ejemplo.

#include<bits/stdc++.h>
using namespace std;

struct  node{
	int data;
	node* lc;
	node* rc;
};
const int maxn=50;
int per[maxn],post[maxn],in[maxn],n,num=0;
node* create(int postl,int postr,int inl,int inr){
	if(postl>postr)return NULL;
	node* root=new node;
	root->data=post[postr];
	int k;//新的根节点
	for(k=inl;k<=inr;k++){
		if(in[k]==post[postr])break;
	}
	int numleft=k-inl;//左子树数量
	root->lc=create(postl,postl+numleft-1,inl,k-1);
	root->rc=create(postl+numleft,postr-1,k+1,inr);
	return root;
}
void BFS(node* root){
	queue<node*> q;//层次遍历使用queue
	q.push(root);
	while(!q.empty()){
		node* now=q.front();
		q.pop();
		printf("%d",now->data);
		num++;
		if(num<n)printf(" ");
		if(now->lc!=NULL)q.push(now->lc);
		if(now->rc!=NULL)q.push(now->rc);
	}
}
int main(){
	cin>>n;
	for(int i=0;i<n;i++)scanf("%d",&post[i]);
	for(int i=0;i<n;i++)scanf("%d",&in[i]);
	node* root=create(0,n-1,0,n-1);
	BFS(root);
    return 0;
}