PAT甲级真题-1020 Tree Traversals详解优化

1020 Tree Traversals

题目链接

https://pintia.cn/problem-sets/994805342720868352/problems/994805485033603072

解题思路

典型的由二叉树的中序和先序序列构建二叉树；之前写模板的时候用的指针构建的二叉树，这次是用静态数组来写的，思路是一样的，都是递归找每个子树的根节点。

总结

1.关于二叉树的详细解释：

https://blog.csdn.net/alpha_xia/article/details/108442469

2.用静态数组的一个坑点是，Tree[]数组一定要开很大，极端情况下才能防止段错误；

代码展示

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

*/
int Tree[155];
int posttre[50];
int intre[50];
int k;
void CreateByPostAndIn(int rott,int pl,int pr,int il,int ir){
	if(pl>pr){
		return;
	}
//	cout<<pl<<" "<<pr<<" "<<il<<" "<<ir<<endl;
	int rot=posttre[pr]; Tree[rott]=rot;
	int rotid=0,leftnum=0;
	for(int i=il;i<=ir;i++){
		if(intre[i]==rot){
			rotid=i;
			break;
		}
	}
	leftnum=rotid-il;
	CreateByPostAndIn(2*rott,pl,pl+leftnum-1,il,rotid-1);
	CreateByPostAndIn(2*rott+1,pl+leftnum,pr-1,rotid+1,ir);
}
void layerorder(int rot){
	queue<int>q;
	q.push(rot);
	while(!q.empty()){
		int curr=q.front();
	     	q.pop();
	    if(curr!=rot)cout<<" ";
		cout<<Tree[curr];
		if(Tree[2*curr]!=0){
			q.push(2*curr);	
		}
		if(Tree[2*curr+1]!=0){
			q.push(2*curr+1);	
		}
	}
}
int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>posttre[i];
	for(int i=1;i<=n;i++)
		cin>>intre[i];
	CreateByPostAndIn(1,1,n,1,n);
	//cout<<"A"<<endl;
	layerorder(1);
	//cout<<endl<<Tree[1]<<endl;
	return 0;
}