题目还是kmp做法 由于北大可能最近在比赛。。。先做了几道HDU的题目
题干大意:
给 t 组数据,每个数据前先输入 字符串 S 的长度 l 之后输入 S
求字符串内 前缀数 + 和前缀匹配的子串数
举例
abab 前缀有 a ab aba abab 匹配数 为 2
故输出 6
对于这个问题。。更加考验对于 next 数组的理解了。
首先我们看一下 任意一个子串 以 ababc为例吧
首先 我们看一下 ababc 的 next 数组 : -1 0 0 1 2 0
next 数组数字连续增长 有一个特殊的意义 那就是代表这前边存在一个连续的前缀对应上了从 0 开始增长的这一串 后缀,那么 当next 数组 不再是 +1 这种形式增长的时候,就代表着,连续相同的前缀和后缀对应结束了。
譬如 aba b中 我们看得见 next [3] = 1,代表前三位里最起码有一个匹配上的 前缀和后缀,而一旦 c[3] 在 kmp 扫描时候也成功匹配,
next[4] = next [3] + 1 ,代表着又出现一对匹配的前缀和后缀,那么当出现一个“断点的时候”,就代表 匹配结束了那么 结尾的出现的 next [ len ] 就是到最后出现的可以匹配的前缀后缀数,我们也需要计算上,
所以综上 我们需要计算匹配子串数目 = next [ len ] + 不符合 next [ i ] + 1 == next [ i + 1 ] 的 next [ i ]
而前缀数目。。 从开始一直挨个数到结尾 一共就为 len 种
所以 可以解决这个问题了
以下 为 ac 代码
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int maxn = 1e6+5;
const int mod = 1e4+7;
int nex[maxn];
int len;
char c[maxn];
void getnex()
{
int l = strlen(c);
int i=0,j=-1;
nex[0]=-1;
while(i<l)
{
if(j==-1 || c[i]==c[j])
nex[++i] = ++j;
else
j= nex[j];
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%s",&len , c);
getnex();
int l = strlen(c);
int res = 0;
for(int i=0;i<l;i++)
{
if(nex[i]!=0 && nex[i] + 1 != nex[i+1])
res += nex[i];
}
for(int i=0;i<=l;i++)
printf("%d ",nex[i]);
printf("\n");
printf("%d\n",(res+l+nex[l])%mod);//别忘了取模
}
return 0;
}