Suppose that all the keys in a binary tree are distinct positive integers. Given the preorder and inorder traversal sequences, you are supposed to output the first number of the postorder traversal sequence of the corresponding binary tree.
Input Specification:
Each input file contains one test case. For each case, the first line gives a positive integer N (≤ 50,000), the total number of nodes in the binary tree. The second line gives the preorder sequence and the third line gives the inorder sequence. All the numbers in a line are separated by a space.
Output Specification:
For each test case, print in one line the first number of the postorder traversal sequence of the corresponding binary tree.
Sample Input:
7
1 2 3 4 5 6 7
2 3 1 5 4 7 6
Sample Output:
3
题意:先序和中序序列求后序序列第一个数字
思路:用柳神的方法连建树都不用建了,参数为prel,inl,inr,设置了一个flag变量来纪录有没有输出后序序列的第一个数字,如果输出了flag就为true。
递归终止条件inl>inr||flag=true
其他语句为先得出先序的第一个数在中序的索引,然后执行递归,判断如果flag=false就打印in[l]。一个比较坑的点是,vector在全局定义的时候,在主函数中要重新定义它的大小
#include<iostream>
#include<cstdio>
#include<stack>
#include<queue>
#include<vector>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<set>
#include<map>
#include<cctype>
#include<cstdlib>
#include<ctime>
using namespace std;
bool flag = false;
vector<int> pre,in;
void postorder(int prel,int inl,int inr) {
if(inl > inr || flag == true) return;
int i = inl;
while(in[i] != pre[prel]) i++;
postorder(prel + 1,inl,i - 1);
postorder(prel + i - inl + 1,i + 1,inr);
if(flag == false) {
printf("%d",in[i]);
flag = true;
}
}
int main() {
int n;
scanf("%d",&n);
pre.resize(n);
in.resize(n);
for(int i = 0;i < n;i++) scanf("%d",&pre[i]);
for(int i = 0;i < n;i++) scanf("%d",&in[i]);
postorder(0,0,n - 1);
return 0;
}