hdu3666 Count the string

题意：给你一个字符串，例如abab,他的前缀有a,ab,aba,abab,求这些前缀在串中能被匹配的次数。abab 就是 2+2+1+1=6.这题数据很弱，例如 5 aaaaa的数据应该是15 8 abababab应该是 20

思路：数据很弱，所以可以暴力（O^2），做法是先扫一遍与第一个相同的位置记录下来，将记录的位置作为起点比较串下一位，相同就记录，依次从串的第一个到最后一个。

#include<bits/stdc++.h>
using namespace std;
const int maxn=200000+10,mod=10007;
char s[maxn];
int main()
{
int T;scanf("%d",&T);
while(T--){
int len;
scanf("%d %s",&len,s);
int b[len+1];
int t=0,ans=0;
for(int i=0;i<len;i++){
if(s[i]==s[0]){
ans=(ans+1)%mod;
b[t++]=i;
}
}
for(int i=1;i<len;i++){
int temp=t;
t=0;
for(int j=0;j<temp;j++){
if(s[b[j]+1]==s[i]){
b[t++]=b[j]+1;
ans=(ans+1)%mod;
 }
}
}
printf("%d\n",ans%mod);
}
}

借助kmp算法的next[]数组可以优化到O(n),next[i]表示前i-1位的子串出现前后缀相等的位置。我们设num[i]表示前-1子串里前缀出现匹配的个数。

我们有转移方程： num[i]=num[next[i]]+1;

#include<bits/stdc++.h>
using namespace std;
const int maxn=200010,mod=10007;
int Next[maxn],num[maxn];
char s[maxn];
void get_next(char *s,int *nxt,int len)
{
memset(nxt,0,sizeof(nxt));
nxt[0]=-1;
int i=0,j=-1;
while(i<len){
if(j==-1||s[i]==s[j]) nxt[++i]=++j;
else j=nxt[j];
}
}

int main()
{
int T;scanf("%d",&T);
while(T--){
int len;scanf("%d",&len);
scanf("%s",s);
get_next(s,Next,len);
memset(num,0,sizeof(num));
int ans=0;
for(int i=1;i<=len;i++){
num[i]=(num[Next[i]]+1)%mod;
ans=(ans+num[i])%mod;
}
printf("%d\n",ans);
}
}