第05次CCF计算机软件能力认证《最佳文章》AcWing 3221. 最佳文章题解

 思路：

构造一颗字典树，并且在维护end数据时做了优化，将节点u的end额外加上了他的fail指向的节点的end，这是利用了字典树fail指向的节点与当前节点的后缀关系，设该节点为u，fail指向的节点为v，则节点0~v表示的字符串一定是0~u的后缀，那么如果递推时新增了这个字母一定可以使文章重要度增加end[u]

容易发现，

1.最优解字符串一定只包含字典树上的字母，

2.最优解的首个字母一定是字典树第一层出现的字母

3.最优解从某个字母起一直到最后一个字母一定是字典树从树根开始按顺序到其中某一个节点（注意，这颗字典树在求fail时其实是被修改过的，某一个节点不存在的儿子是会被接回树根的，具体看代码），也就是说递推时，每新增一个字母，一定是顺着当前最优解某个后缀在字典树中按照顺序向下延伸的，那么每向下延伸到一个新节点，文章可以增加的重要度就是他的end，那么往哪条边走才是新增一个字符后的最优解？那当然是选end最大的那个节点

那么问题可以转换成，有一个小人从0节点出发，一共跳m次，每次只能沿着字典树的边跳，每经过一个节点u，就可收获end[u],求小人的最大收获

而且注意不能往回跳，具体体现在代码初始化矩阵中 Init() 都是枚举从当前节点出发往他的儿子跳

维护一个矩阵a[][],a[i][j]表示假如当前最优解下小人在节点i，那么他往节点j跳可以得到的收获

那么他跳两次从i到k的最大收获是

不妨将这个运算当做*的重载

故跳m次的最大收获就是将状态转移矩阵进行m次定义的*运算后

可以理解为枚举找出小人从0节点出发，跳m次在t节点结束的所有可能收获，选取其中的最大值

需要注意的是如果节点i与j没有边相连，那么初始化矩阵a时a[i][j]要附一个极小值，表示从i跳到j非法，因为字典树中只有按顺序往下走得到的才是出现过的单词

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f3f3f3f3f;
int n,cnt;
ll m;
char str[109];
struct info
{
	int son[30],fail,end;
}trie[109];
struct matrix
{
	ll num[109][109];
	matrix()
	{
		for (int i=0;i<109;i++)
			for (int j=0;j<109;j++)
				num[i][j]=-inf;
	}
}ans;
matrix operator * (matrix a,matrix b)
{
	matrix res;
	memset(res.num,0xaf,sizeof(res.num));
	for (int i=0;i<=cnt;i++)
		for (int j=0;j<=cnt;j++)
			for (int k=0;k<=cnt;k++)
				if (a.num[i][k]+b.num[i][k]>=0)
					res.num[i][j]=max(res.num[i][j],a.num[i][k]+b.num[k][j]);
	return res;
}
matrix Quickpow(matrix a,ll b)
{
	matrix res=a;
	b--;
	for (;b;b>>=1)
	{
		if (b&1) res=res*a;
		a=a*a;
	}
	return res;
}
void Insert(char *s)
{
	int now=0,ch;
	for (int i=0;s[i];i++)
	{
		ch=s[i]-'a';
		if (!trie[now].son[ch])
			trie[now].son[ch]=++cnt;
		now=trie[now].son[ch];
	}
	trie[now].end++;
}
void Getfail()
{
	queue<int> q;
	for (int i=0;i<26;i++)
		if (trie[0].son[i])
			q.push(trie[0].son[i]);
	for (int u,v;!q.empty();)
	{
		u=q.front();
		q.pop();
		trie[u].end+=trie[trie[u].fail].end;
		for (int i=0;i<26;i++)
		{
			v=trie[u].son[i];
			if (v)
			{
				trie[v].fail=trie[trie[u].fail].son[i];
				q.push(v);
			}
			else
				trie[u].son[i]=trie[trie[u].fail].son[i];
		}
	}
}
void Init()
{
	for (int i=0,u;i<=cnt;i++)
	{
		for (int j=0;j<26;j++)
		{
			u=trie[i].son[j];
			ans.num[i][u]=trie[u].end;
		}
	}
}
void Solve()
{
	ans=Quickpow(ans,m);
	ll MAX=ans.num[0][0];
	for (int i=1;i<=cnt;i++)
		MAX=max(MAX,ans.num[0][i]);
	cout<<MAX;
}
int main()
{ 
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n>>m;
	memset(trie,0,sizeof(trie));
	for (int i=1;i<=n;i++)
	{
		cin>>str;
		Insert(str);
	}
	Getfail();
	Init();
	Solve();
	return 0;
}