PAT1138. Postorder Traversal (25) 已知树的前序和中序，求树的后序的第一个节点（递归）

/*************************
题意:
给出前序和中序遍历
求后序遍历的第一个点
************************/
/***********************
解题思路：
用递归
先在前序里找出本次递归子树的root
然后在中序里找出root的位置
如果有左树，则去左树里递归
如果没有左树，则去右树里递归
递归到第一个叶子，则那个就是后序的第一个点

*************************/
/***********************
笔记：
*********************/
#include<iostream>
#include<stdio.h>
#include<string>
#include<vector>
#include<queue>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<stack>
#include<map>
#include<set>
#include<unordered_map>
using namespace std;
#define M 50005
#define INF 0x7ffffff

int pre[M],in[M];

int getpost(int pl,int pr,int il,int ir){
	int root = pre[pl];
	int i;
	if(pl == pr)
		return pre[pl];
	for(i=il;i<=ir;i++){
		if(root == in[i])
			break;
	}
	int len = i - il; //左子树中节点的个数
	if(i!=il){  //左边有左子树，则插入左边
		return getpost(pl+1, pl+len,il,i-1);
	}
	else{  //没有左子树，则插入右边
		return getpost(pl+len+1,pr,i+1,ir);
	}
}

int main(){
	int n, i, k;
	cin>>n;
	for(i=0;i<n;i++)
		cin>>pre[i];
	for(i=0;i<n;i++)
		cin>>in[i];
	cout<<getpost(0, n-1, 0, n-1)<<endl;

	return 0;
}