【后序+中序重构】PAT A1020 Tree Traversals (25分)

题目描述

知识点

二叉树的遍历

我的实现

码前思考

我第一次用的是建树的方式，后来看了柳神的代码，原来可以之间得到层序遍历结果，妙啊。

利用的是完全二叉树的性质。

代码实现

#include <iostream>
#include <queue>
#include <map>
#include <algorithm>
using namespace std;
const int maxn = 40; 

int post[maxn];
int in[maxn];
map<int,int> layer;

void create(int postL,int postR,int inL,int inR,int idx){
	if(postL>postR){
		return;
	}else{
		layer[idx] = post[postR];
		int k;//k表示根节点在中序序列中的位置
		for(k=inL;k<=inR;k++){
			if(layer[idx] == in[k]){
				break;
			}
		}
		
		//得到左子树的长度
		int numLeft = k-inL;
		
		create(postL,postL+numLeft-1,inL,k-1,2*idx);
		create(postL+numLeft,postR-1,k+1,inR,2*idx+1);
		
		return; 
	}
}

int main(){
	int n;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&post[i]);
	}
	for(int i=0;i<n;i++){
		scanf("%d",&in[i]);
	}
	
	create(0,n-1,0,n-1,1);
	
	map<int,int>::iterator it = layer.begin();
	while(true){
		printf("%d",it->second);
		it++;
		if(it!=layer.end()){
			printf(" ");
		}else{
			break;
		}
	}
	
	return 0;
}

码后反思

我发现PAT好多这种题可以直接得到结果，而不需要建树。

尤其是体现在这种通过前序，中序和后序遍历的题目上。