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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Mr0cheng/article/details/79497986
  1. 和一般的中序和前序搭配建树不同,这里一定要分清每次递归要做的子任务。就是将当前preOrder的第一个看成根节点,然后去划分根节点的左右子树(因为NLR)。
  2. 而且,如果当前的区间只有两个节点(其中一个看成根节点),又因为只有一个孩子节点,故无法判断该孩子节点是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;
}

猜你喜欢

转载自blog.csdn.net/Mr0cheng/article/details/79497986
今日推荐