F - Mr. Panda and Fantastic Beasts Gym - 101194F

題解:對於每個字符串後面加一個特殊符號把所有的串連接在一起,然後在後綴數組中找到後綴的位置出現在第一個串,對於這個串找到前後最近的不是第一個串的字符串,取height數組前後區間最小值的最大值,如果後綴小於第一個串中以i爲起點的後綴長度,取最小值,如果最小值相同,取rak最小的。

代碼:

#include <bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxn=3e5+100;
int t1[maxn],t2[maxn],c[maxn],num[maxn];
int rak[maxn],height[maxn],sa[maxn];
char str[maxn];
int m;
bool cmp(int *r,int a,int b,int I){
    return r[a]==r[b]&&r[a+I]==r[b+I];
}

void da(char str[],int n){
    m=1000;
    n++;
    int i,j,p,*x=t1,*y=t2;

    for(i=0;i<m;i++) c[i]=0;

    for(i=0;i<n;i++) c[x[i]=str[i]]++;

    for(i=1;i<m;i++) c[i]+=c[i-1];

    for(i=n-1;i>=0;i--) sa[--c[x[i]]]=i;

    for(j=1;j<=n;j<<=1){
        p=0;
        for(i=n-j;i<n;i++) y[p++]=i;

        for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;

        for(i=0;i<m;i++) c[i]=0;

        for(i=0;i<n;i++) c[x[y[i]]]++;

        for(i=1;i<m;i++) c[i]+=c[i-1];

        for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1; x[sa[0]]=0;
        for(i=1;i<n;i++)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j) ? p-1:p++;
            if(p>=n) break;
            m=p;
    }
    int k=0;
    n--;
    for(i=0;i<=n;i++) rak[sa[i]]=i;
    for(i=0;i<n;i++) {
        if(k) k--;
        j=sa[rak[i]-1];
        while(str[i+k]==str[j+k]) k++;
        height[rak[i]]=k;
    }
}
int anss[maxn],rax[maxn],px[maxn]; char tmp[maxn];
int main()
{
    int T,n,cas=0;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        int last=0,ax=0; ++cas;
        for(int i=1;i<=n;i++){
            scanf("%s",tmp);
            int len=strlen(tmp);
            for(int j=0;j<len;j++){
                str[last++]=tmp[j];
            }
            px[++ax]=last-1;
            str[last++]='#';
        }
        str[last]=0;
        da(str,last);
        for(int i=0;i<=px[1];i++) anss[i]=0;
        int pxx=inf;
        for(int i=1;i<=last;i++){
            if(sa[i]<=px[1]) {
                rax[sa[i]]=i;
                if(pxx==inf) continue;
                if(pxx!=inf){
                    pxx=min(pxx,height[i]);
                }
                anss[sa[i]]=max(anss[sa[i]],min(pxx,px[1]-sa[i]+1));
            }else if(str[sa[i]]!='#') pxx=height[i+1];
        }
        pxx=inf;
        for(int i=last;i>=1;i--){
            if(sa[i]<=px[1]) {
                if(pxx==inf) continue;
                anss[sa[i]]=max(anss[sa[i]],min(pxx,px[1]-sa[i]+1));
                if(pxx!=inf)
                    pxx=min(pxx,height[i]);
            }else if(str[sa[i]]!='#') pxx=height[i];
        }
        int anx=inf;
        for(int i=0;i<=px[1];i++){
            if(anss[i]==px[1]-i+1) continue;
            anx=min(anx,anss[i]);
        }
        if(anx==inf){
            printf("Case #%d: Impossible\n",cas); continue;
        }
//        for(int i=0;i<=px[1];i++) cout<<rak[i]<<endl;
        int flag,op=inf;
        for(int i=0;i<=px[1];i++){
            if(rax[i]<op&&anx==anss[i]&&anss[i]!=px[1]-i+1) {
                flag=i; op=rax[i];
            }
        }
        printf("Case #%d: ",cas);
        for(int i=flag;i<=anx+flag;i++) printf("%c",str[i]);
        printf("\n");
    }
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章