给出一个字符串,求出其所有前缀在当前串中出现的次数和
我刚开始的做法是枚举所有的前缀,用kmp判断匹配
数据量200,000。时间复杂度O(4e10),毫无疑问TLE了
后来才知道是利用next性质来做
对于abab
我们从下标1开始
则很容易得到
next[1] = 0
next[2] = 0
next[3] = 1
next[4] = 2
令dp[i]为以p[i]结尾的前缀个数
根据next数组含义,我们很容易得到dp[i] = dp[str[i]]+1;
如对于abab
dp[1]对应于:a(前面的)
dp[2]对应于:ab(前面的)
dp[3]对应于:a(后面的), aba
dp[4]对应于:ab(后面的), abab
故和为6个
感觉这个题比较蛋疼的地方就是求next要从下标1开始。。。由于对kmp理解的还不够透彻,我只会从下标0开始
代码如下:
#include <cstdio>
#include <cstring>
#define MAXN 200100
#define LL long long
#define MOD 10007
using namespace std;
int n;
int next[MAXN], dp[MAXN];
char str[MAXN];
void get_next(char p[]) {
next[1] = 0;
int i, j = 0;
for(int i=2; i<=n; ++i) {
while(j>0 && str[j+1]!=str[i])
j = next[j];
if(str[j+1] == str[i])
++j;
next[i] = j;
}
}
int main(void) {
int T;
scanf("%d", &T);
while(T--) {
scanf("%d", &n);
scanf("%s", str+1);
get_next(str+1);
int sum = 0;
dp[0] = 0;
for(int i=1; i<=n; ++i) {
dp[i] = dp[next[i]]+1;
sum = (sum+dp[i])%MOD;
}
printf("%d\n", sum);
}
return 0;
}