Tree Traversal again

题目链接

根据前序和中序写出后序
—可以创建出树再后序遍历
—也可以不创建出树直接把后序遍历的结构存到数组
两种代码差不多，之前写的并没有理解，重新写一遍
要注意一些细节，以及递归出口有两个。

#include<stack>
#include<iostream>
#include<cstring>
using namespace std;
int pre[50],in[50],post[50];
int flag=0;
void sol(int pre_f,int in_f,int post_f,int n){
	if(!n) return;
	if(n==1){
		post[post_f]=pre[pre_f];
		return;
	} 
	int root=pre[pre_f];
	post[post_f+n-1]=root;
	int i;
	for(i=0;i<n;i++){
		if(in[in_f+i]==root) break;
	}
	
	int left,right;
	left=i;right=n-left-1;
	sol(pre_f+1,in_f,post_f,left);
	sol(pre_f+left+1,in_f+left+1,post_f+left,right);
}

int main(){
	int n,m,a=0,b=0;cin>>n;
	string op;
	stack<int>s;
	for(int i=0;i<2*n;i++){
		cin>>op;
		if(op[1]=='u'){
			cin>>m;
			pre[a++]=m;
			s.push(m); 
		}
		else {
			in[b++]=s.top() ;
			s.pop() ;
		}
	}
	sol(0,0,0,n);
	for(int i=0;i<n;i++){
		if(flag) cout<<" ";
		cout<<post[i];
		flag=1;
	}
	return 0;
}

一道类似的题目，根据前序和中序创建树—PTA-41
题目链接

//先创建出树，之后遍历得到高度 
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cstdlib>
using namespace std;
char a[55];
char b[55];//全局变量嘛 
 
typedef struct Tree{
	int data;
	struct Tree* left;
	struct Tree* right;
}Tree,*BinT;


int H(BinT T){
	int hl,hr,maxh;
	if(T){
		hl=H(T->left);
		hr=H(T->right);
		maxh=hl>hr?hl:hr;
		return (maxh+1);
	}
	else return 0;//空树的高度为0 
}

BinT creat_Tree(int a_f,int b_f,int n){
	BinT T;
	int i=0;
	if(!n) return NULL;
	T=new Tree();
	T->data =a[a_f];
	if(n==1) return T;
	
	for(i=0;i<n;i++){
		if(b[b_f+i]==a[a_f]) break;
	}

    int left,right;
    left=i;right=n-left-1;
	T->left =creat_Tree(a_f+1,b_f,left);
	T->right =creat_Tree(a_f+left+1,b_f+left+1,right);
	return T;
} 


int main(){
	int n;
	cin>>n>>a>>b;
	BinT T=creat_Tree(0,0,n);
	cout<<H(T)<<endl;
	return 0;
}