题意:给定一字符串,求它所有的前缀在该字符串中出现的次数的和。
思路:
单纯的跑n遍kmp肯定TLE
我们考虑next数组的本质
对于任意位置i的next[i]的值,令len=next[i],如果len不等于0,说明0~i这一子串中(0到k-1 和i-k+1到i相等) 即0到k-1这一前缀子串 出现了两次
那我们直接就判断 对于任意i 如果他的next[i]不为0 则对应前缀出现了2次 否则只出现一次
#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=2e5+5;
int Next[maxn];
void getnext(char *ptr)
{
int i,n,k;
n=strlen(ptr);
k=-1;
Next[0]=-1;
i=0;
while(i<n)
{
if(k==-1 || ptr[i]==ptr[k])
{
i++;
k++;
Next[i]=k;
}
else k=Next[k];
}
}
int mod=10007;
char s[maxn];
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int len;
scanf("%d%s",&len,s);
getnext(s);
int ans=len;
for(int i=1;i<=len;i++)
{
if(Next[i])
ans++;
}
printf("%d\n",ans%mod);
}
}