PAT A1119 前序遍歷後續遍歷不唯一生成

clipboard.png

之前僅僅接觸過如何通過二叉樹的中序+先序/後序序列生成唯一二叉樹,這一次見到了這個新的題型;

這裏先梳理一個概念,之所以會生成樹不唯一,一定是有一個葉子,無論其在父節點的左右子節點,都可能生成相同的先序和後序遍歷序列;

所以這個時候,思路就很清晰,我們判別一個序列是否唯一,條件就是是否有一個節點只有一個子節點;

大致的序列分割和先序和後續相同,這個後面專門開一個blog進行總結;

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<vector>
#include<set>
#include<algorithm>
using namespace std;
using std::vector;
using std::set;
vector<int>in,pre,post;
bool uniq=true;

void chargement(int prl,int prr,int pol,int por){
    if(prl==prr){
        in.push_back(pre[prl]);
        return ;
    }
    if(pre[prl]==post[por]){
        int i=prl+1;
        while(i<=prr&&pre[i]!=post[por-1])
            i++;
        if(i-prl>1)
            chargement(prl+1,i-1,pol,pol+(i-prl-1)-1);
        else
            uniq=false;
        in.push_back(post[por]);
        chargement(i,prr,pol+(i-prl-1),por-1);
    }
}


int main(){
    int n;
    scanf("%d",&n);
    pre.resize(n);
    post.resize(n);
    for(int i=0;i<n;i++)
        scanf("%d",&pre[i]);
    for(int i=0;i<n;i++)
        scanf("%d",&post[i]);
    chargement(0,n-1,0,n-1);
    printf("%s\n%d", uniq == true ? "Yes" : "No", in[0]);
    for (int i = 1; i < in.size(); i++)
        printf(" %d", in[i]);
    printf("\n");
    system("pause");
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章