PAT甲级 1119. Pre- and Post-order Traversals

最新推荐文章于 2022-02-26 19:51:19 发布

原创最新推荐文章于 2022-02-26 19:51:19 发布 · 335 阅读

0 ·

CC 4.0 BY-SA版权

PAT 同时被 2 个专栏收录

51 篇文章

订阅专栏

PAT甲级

51 篇文章

订阅专栏

本文介绍了一种通过先序和后序遍历序列构建二叉树的方法，并通过递归算法实现。重点在于如何划分左右子树及判断二叉树是否唯一。

和一般的中序和前序搭配建树不同，这里一定要分清每次递归要做的子任务。就是将当前preOrder的第一个看成根节点，然后去划分根节点的左右子树（因为NLR）。
而且，如果当前的区间只有两个节点（其中一个看成根节点），又因为只有一个孩子节点，故无法判断该孩子节点是NLR中的L还是R。所以当preR-preL==1时，即不是独一无二的。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 30+2;
int preOrder[maxn],postOrder[maxn];
vector<int> inOrder;
bool flag=true;
void dfs(int preL,int preR,int postL,int postR){
    if(preL>preR)return;
    if(preR-preL==1)flag=false;
    int i;
    for(i=postL;i<=postR-1;++i){
        if(postOrder[i]==preOrder[preL+1])break;
    }
    if(preL!=preR)dfs(preL+1,preL+i-postL+1,postL,i);
    inOrder.push_back(preOrder[preL]);
    if(preL!=preR)dfs(preR+i+2-postR,preR,i+1,postR-1);
}

int main(){
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;++i)scanf("%d",preOrder+i);
    for(int i=0;i<n;++i)scanf("%d",postOrder+i);
    dfs(0,n-1,0,n-1);
    if(flag)printf("Yes\n");
    else printf("No\n");
    for(vector<int>::iterator it=inOrder.begin();it!=inOrder.end();++it){
        if(it==inOrder.begin())printf("%d",*it);
        else printf(" %d",*it);
    }
    printf("\n");
    return 0;
}