CSP201509T5最佳文章

最新推荐文章于 2024-08-13 14:56:59 发布

wly127

最新推荐文章于 2024-08-13 14:56:59 发布

阅读量128

点赞数 3

分类专栏： CSP 文章标签：算法

本文链接：https://blog.csdn.net/wly1127/article/details/138012868

版权

CSP 专栏收录该内容

23 篇文章 0 订阅

订阅专栏

文章介绍了如何利用Trie树和矩阵乘法解决一个问题：给定n个字符串，计算在一篇文章中最多能包含多少个长度为m的字符串，同时保证这些字符串是连续的且不重复。算法涉及到动态规划和查找最长公共前缀。

摘要由CSDN通过智能技术生成

题意：有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;
}