[KMP]P2375

zzz阿哲

已于 2022-08-13 11:17:38 修改

阅读量94

点赞数

分类专栏：字符串 # KMP 文章标签：字符串处理 KMP算法前缀后缀子串乘积计算时间复杂度优化

于 2022-07-27 18:21:09 首次发布

本文链接：https://blog.csdn.net/qq_66753825/article/details/126020739

版权

字符串同时被 2 个专栏收录

6 篇文章 0 订阅

订阅专栏

KMP

2 篇文章 0 订阅

订阅专栏

定义字符串的前 $i$ 个字符组成的字符串中一最大子串 $T$ 即使前缀也是后缀，且 $|T|\leq i/2$ ，则定义 $n u m [i] = ∣ T ∣$ ，求 $n u m [i] + 1$ 之积 $m o d$ 1000000007。 $|S|\leq 10^6$ 。

熊猫：“对于字符串S的前i个字符构成的子串，既是它的后缀又是它的前缀的字符串中（它本身除外），最长的长度记作 $n x t [i]$ 。”

根据 $n u m$ 数组的定义， $n u m [i]$ 表示在 $nxt[nxt[...]]\leq i/2$ 之后在跳 $n u m [i]$ 次 $n x t$ 就会等于0

举个例子：

	1	2	3	4	5	6	7	8
s[i]	a	b	a	b	a	b	a	b
nxt[i]	0	0	1	2	3	4	5	6

$n x t [7] = 5$ 由于有交叉， $n x t [7] = n x t [n x t [7]] = n x t [5] = 3$ ,表示 $s [1...3] = s [5...7]$ ， $n u m [7] + +$ 。同理在 $n x t [7] = 3$ 时 $n u m [7] + +$ 。

所以这题只用先跑一遍KMP，之后用 $n x t$ 数组回溯即可。

考虑到时间复杂度有可能退化，所以在每一次 $j$ 指针计数结束后不还原，就可以使之后的 $n x t$ 更快接近 $i /2$ 。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int n,nxt[1000005],num[1000005],slen;
char s[1000005];
ll ans;

int main()
{
	scanf("%d",&n);
	while(n)
	{
		n--;
		memset(num,0,sizeof(num));
		ans=1;
		scanf("%s",s+1);
		slen=strlen(s+1);
		num[1]=1;
		for(int i=2,j=0;i<=slen;++i)
		{
			while(j>0 && s[i]!=s[j+1])
				j=nxt[j];
			if(s[i]==s[j+1])
				++j;
			nxt[i]=j;
			num[i]=num[j]+1;
		}
		for(int i=2,j=0;i<=slen;++i)
		{
			while(s[i]!=s[j+1] && j>0)
				j=nxt[j];
			if(s[i]==s[j+1])
				++j;
			while(j*2>i)
				j=nxt[j];
			ans=ans*(num[j]+1)%1000000007;
		}
		printf("%lld\n",ans);
	}
}