题意:有n个字符串,求一篇由m个字母组成的文章最多可以包含多少个字符串。
#include<bits/stdc++.h>
using namespace std;
struct node
{
int son[26];
int flag,fail;
}trie[250];
char s[200001];
long long ans[200001];//以j结点结尾的模式串数量
long long n,m,cnt=1;
struct matrix
{
long long s[150][150];
matrix()
{
memset(s,-0xf,sizeof s);
}
}A;
matrix operator *(const matrix &a,const matrix &b)
{
matrix c;
for(int i=1;i<=cnt;i++)
for(int j=1;j<=cnt;j++)
for(int k=1;k<=cnt;k++)
c.s[i][j]=max(c.s[i][j],a.s[i][k]+b.s[k][j]);
return c;
}
matrix pow(matrix a,long long p)
{
matrix t;
for(int i=1;i<=cnt;i++)t.s[i][i]=0;
while(p)
{
if(p&1)t=t*a;
a=a*a;
p>>=1;
}
return t;
}
void insert(char* s)
{
int u=1,len=strlen(s);
for(int i=0;i<len;i++)
{
int v=s[i]-'a';
if(!trie[u].son[v])trie[u].son[v]=++cnt;
u=trie[u].son[v];
}
ans[u]++;
trie[u].flag++;//标志结束
}
queue <int> q;
void getFail()
{
for(int i=0;i<26;i++)trie[0].son[i]=1;//初始化0的所有儿子都是1
q.push(1);
trie[1].fail=0; //将根的fail为0
while(!q.empty())
{
int u=q.front();q.pop();
ans[u]+=ans[trie[u].fail];
for(int i=0;i<26;i++) //遍历所有儿子
{
int v=trie[u].son[i];
int Fail=trie[u].fail; //trie[Fail].son[i]就是和v值相同的点
if(!v){trie[u].son[i]=trie[Fail].son[i];continue;} //不存在该节点
trie[v].fail=trie[Fail].son[i];
q.push(v);
}
}
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>s;
insert(s);
}
getFail();
for(int i=1,p;i<=cnt;i++)
for(int j=0;j<26;j++)
{
for(p=i;p&&!trie[p].son[j];p=trie[p].fail);//从trie中找到26个字母作为下一个
p=p?trie[p].son[j]:1;
A.s[i][p]=ans[p];//矩阵i,j代表从结点i到结点j最大数量
}
A=pow(A,m);
long long sum=0;
for(int i=1;i<=cnt;i++)sum=max(sum,A.s[1][i]);//从1(根结点)到每个结点找最大
cout<<sum<<endl;
return 0;
}