1. 程式人生 > >洛谷 P1368 工藝 後綴自動機

洛谷 P1368 工藝 後綴自動機

cst cpp 後綴 return urn stdin ins cstring efi

後綴自動機沙茶題
說實話還是沒有那麽理解後綴自動機.

Code:

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <map>
#define setIO(s) freopen(s".in","r",stdin) 
#define maxn 1000000 
using namespace std;
int n,arr[maxn],m,last=1,tot=1,len[maxn],f[maxn];  
map<int,int>ch[maxn]; 
map<int,int>::iterator  it;
void ins(int c){
    int p=last,np=last=++tot,q;
    len[np]=len[p]+1;
    while(p&&!ch[p][c]) ch[p][c]=np,p=f[p]; 
    if(!p) f[np]=1;
    else{
        q=ch[p][c];
        if(len[q]==len[p]+1)f[np]=q;
        else
        {
            int nq=++tot;
            len[nq]=len[p]+1;
            ch[nq]=ch[q];
            f[nq]=f[q];
            f[np]=f[q]=nq;
            while(p&&ch[p][c]==q)
                ch[p][c]=nq,p=f[p];
        }
    }
}
int main(){
    //setIO("input"); 
    scanf("%d",&n),m=2*n; 
    for(int i=1;i<=n;++i) scanf("%d",&arr[i]);
    for(int i=n+1;i<=m;++i) arr[i]=arr[i-n]; 
    for(int i=1;i<=m;++i) ins(arr[i]); 
    int p=1;
    while(n -- )
    {
        it=ch[p].begin();
        printf("%d" , it->first);
        if(n) printf(" ");
        p = it->second;
    }
    return 0; 
}

  

洛谷 P1368 工藝 後綴自動機