OJ-Problems-Source/HDOJ/2243_autoAC.cpp

213 lines
4.3 KiB
C++

#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
#define LL unsigned __int64
struct node
{
int cnt;
struct node *fail;
struct node *next[26];
struct node *jump[26];
};
node root[33];
int num;
char str[10];
LL mod=10330176681277348905;
void insert(char word[])
{
int i=0;
node *tmp=root;
while(word[i])
{
int b=word[i]-'a';
if(tmp->next[b]==NULL)
{
tmp->next[b]=root+num;
memset(root+num,0,sizeof(struct node));
num++;
}
tmp=tmp->next[b];
i++;
}
tmp->cnt=1;
}
node *q[33];
int head,tail;
void add_fail()
{
head=tail=0;
q[tail++]=root;
while(head<tail)
{
node *x=q[head++];
for(int i=0;i<26;i++)
{
node *t=x->fail;
while(t!=NULL && t->next[i]==NULL)
t=t->fail;
if(x->next[i]!=NULL)
{
q[tail++]=x->next[i];
if(t==NULL)
x->next[i]->fail=root;
else
{
x->next[i]->fail=t->next[i];
if(t->next[i]->cnt)
x->next[i]->cnt=1;
}
x->jump[i]=x->next[i];
}
else
{
if(t==NULL)
x->jump[i]=root;
else
x->jump[i]=t->next[i];
}
}
}
}
int m;
int n;
struct Mat
{
LL a[2*33][2*33];
void init()
{
int i,j;
for(i=0;i<2*33;i++)
for(j=0;j<2*33;j++)
a[i][j]=0;
}
};
int len;
Mat e;
Mat mul(Mat a,Mat b)
{
Mat res;
int i,j,k;
for(i=0;i<2*len;i++)
{
for(j=0;j<2*len;j++)
{
res.a[i][j]=0;
for(k=0;k<2*len;k++)
{
if(a.a[i][k]>0 && b.a[k][j]>0)
{
res.a[i][j]+=a.a[i][k]*b.a[k][j];
}
}
}
}
return res;
}
LL solve(Mat a,int k)
{
Mat b,ans=e;
int i,j;
b.init();
for(i=0;i<len;i++)
for(j=0;j<len;j++)
b.a[i][j]=a.a[i][j];
for(i=0;i<len;i++)
{
for(j=len;j<2*len;j++)
b.a[i][j]=(i==j-len);
}
for(i=len;i<2*len;i++)
{
for(j=len;j<2*len;j++)
b.a[i][j]=(i==j);
}
k++;
while(k)
{
if(k&1)
{
ans=mul(ans,b);
}
b=mul(b,b);
k/=2;
}
LL res=0;
for(i=0;i<len;i++)
{
res+=ans.a[i][len];
}
return res-1;
}
LL sum(LL a,int k)
{
Mat b;
Mat ans=e;
b.init();
b.a[0][0]=26;b.a[0][1]=1;
b.a[1][0]=0; b.a[1][1]=1;
k++;
while(k)
{
if(k&1)
{
ans=mul(ans,b);
}
b=mul(b,b);
k/=2;
}
return ans.a[0][1]-1;
}
int main()
{
int i,j,k;
for(i=0;i<2*33;i++)
for(j=0;j<2*33;j++)
e.a[i][j]=(i==j);
while(scanf("%d%d",&n,&m)==2)
{
num=1;
memset(root,0,sizeof(struct node));
for(i=0;i<n;i++)
{
scanf("%*c%s",str);
insert(str);
}
add_fail();
int id=0,id1;
Mat res;
for(i=0;i<num;i++)
{
if(root[i].cnt==0)
{
id1=0;
for(j=0;j<num;j++)
{
if(root[j].cnt==0)
{
LL count=0;
for(k=0;k<26;k++)
{
if(root[j].jump[k]==root+i)
{
count++;
}
}
res.a[id][id1]=count;
id1++;
}
}
id++;
}
}
len=id;
LL all=sum(26,m);
LL left=solve(res,m);
all-=left;
if(all<0)
all+=mod;
printf("%I64u\n",all);
}
return 0;
}