PAT 1138 Postorder Traversal (25分)

最新推荐文章于 2022-10-27 16:36:25 发布

ChasingTheFreeWind

最新推荐文章于 2022-10-27 16:36:25 发布

阅读量144

点赞数

分类专栏： PAT 二叉树

本文链接：https://blog.csdn.net/q1072118803/article/details/106095053

版权

PAT 同时被 2 个专栏收录

26 篇文章 0 订阅

订阅专栏

二叉树

12 篇文章 0 订阅

订阅专栏

在这里插入图片描述

题意

已知前序中序求后序第一个数字，递归就完事儿。

代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#define MAX 50010
using namespace std;
int pre[MAX];
int in[MAX];
int ans=-1;
int N;
void getPost(int pL,int pR,int iL,int iR)
{
    if(ans>0)
        return;
    if(pL>pR)
        return;
    int root = pre[pL];
    int pos = find(in,in+iR+1,root)-in;
    int cnt = pos-iL; 
    getPost(pL+1,pL+cnt,iL,iL+cnt-1);
    getPost(pL+cnt+1,pR,iL+cnt+1,iR);
    if(ans<0)
    {
        printf("%d\n",root);
        ans = 0;
        return;
    }
}
int main()
{
    scanf("%d",&N);
    for(int i=0;i<N;i++)
        scanf("%d",&pre[i]);
    for(int i=0;i<N;i++)
        scanf("%d",&in[i]);
    getPost(0,N-1,0,N-1);
    return 0;
}