【GDKOI2017模拟1.21】Rhyme

Description

由于多次交换邮票没有满足所有人的需求，小Z被赶出了集邮部。无处可去的小Z决定加入音乐部，为了让音乐部的人注意到自己的才华，小Z想写一首曲子。为了让自己的曲子更好听，小Z找到了一些好听曲子作为模板。曲谱可以表示成只包含小写字母的字符串，小Z希望自己最终的曲谱中任意一个长度为K的子串都是一个模板的子串。现在小Z想知道自己的曲谱最长可以是多长，如果可以无限长的话请输出INF。

Solution

要前后缀匹配能想到的：sa，sam，kmp，exkmp，ac自动机，骗分神器hash。
经过筛选之后，决定用hash
很显然的想法，只有有两个字符串前后缀重叠≥k-1。
那么可以hash完之后用hash值连那个字符串的节点。那个字符串的节点就变成了中介点。
但是并不需要中介点，只用[l,l+k-1]的字符串向[l+1,l+k]的字符串连边就可以了。
然后就需要找正环。
比赛的时候打了spfa，结果超时。
后来发现只是个DAG，直接用拓扑排序，然后直接向后扫就可以了。
hash的时候直接开unsigned long long 然后让hash自然溢出。

Code

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fod(i,a,b) for(i=a;i>=b;i--)
#define rep(i,a) for(i=first[a];i;i=next[i])
using namespace std;
//typedef long long ll;
typedef unsigned long long ll;
const int maxn=1e5+7,moo=4e5+7;
ll i,j,k,l,t,n,m,len,ans,S;
ll h[moo];
ll first[moo],last[moo],next[moo],num;
ll d[moo],data[moo],ge,hao,in[moo];
ll ci[moo],a[moo],head,tail;
ll x,y,c[maxn],mo,xx,yy,b[moo],mx;
char s[maxn];
bool az,cz,bz[moo];
ll hash(ll x){
    ll y=((x%moo)+moo)%moo;
    while(h[y]!=x&&h[y]){y++;if(y>=moo)y-=moo;}
    h[y]=x;
    return y;
}
void add(ll x,ll y){
    last[++num]=y,next[num]=first[x],first[x]=num;
    in[y]++;
}
void bfs(){
    int head=0,tail=0,i,j,now;
    memset(d,0,sizeof(d));memset(bz,0,sizeof(bz));
    fo(i,1,a[0]){
        if(!bz[a[i]]&&!in[a[i]])bz[a[i]]=1,data[++tail]=a[i];
    }
    while(head<tail){
        now=data[++head];
        rep(i,now){
            in[last[i]]--;
            if(!in[last[i]])data[++tail]=last[i],d[last[i]]=d[now]+1;
        }
    }
}
int main(){
    freopen("rhyme.in","r",stdin);
    freopen("rhyme.out","w",stdout);
//  freopen("fan.in","r",stdin);
//  freopen("fan.out","w",stdout);
    c[0]=1;mo=1000000007;
    fo(i,1,100000)c[i]=c[i-1]*27;
    S=moo;
    while(scanf("%u",&n)!=EOF){
        hao++;
        scanf("%u",&k);num=0;az=1;ge=1;a[0]=0;
        memset(first,0,sizeof(first));
        memset(h,0,sizeof(h));
        memset(in,0,sizeof(in));
        fo(i,1,n){
            scanf("%s",s+1);len=strlen(s+1);
            fo(j,1,len)b[j]=b[j-1]*27+s[j]-'a';
            fo(j,k,len){
                x=b[j-1]-b[j-k]*c[k-1];xx=hash(x);
                y=b[j]-b[j-k+1]*c[k-1];yy=hash(y);
                add(xx,yy);
                if(j==k)a[++a[0]]=xx;
            }
        }
        ans=0;
        bfs();
        fo(i,0,moo-1){
            ans=max(ans,d[i]);
            if(in[i])az=0;
        }
        if(k<=1)az=0;
        if(!az)printf("INF\n");else printf("%u\n",ans+k-1);
    }
}