题意:给一个字符串,计算所有前缀在字符串中出现的次数和。
解法:KMP计算出Next数组后,每个位置的Next数组不断往前递归,每次相应前缀次数就加1.
代码:
/****************************************************** * author:xiefubao *******************************************************/ #pragma comment(linker, "/STACK:102400000,102400000") #include <iostream> #include <cstring> #include <cstdlib> #include <cstdio> #include <queue> #include <vector> #include <algorithm> #include <cmath> #include <map> #include <set> #include <stack> #include <string.h> //freopen ("in.txt" , "r" , stdin); using namespace std; #define eps 1e-8 const double pi=acos(-1.0); typedef long long LL; const int Max=200100; const int INF=10007; int Next[Max]; int num[Max]; char s[Max]; int len=0; void get_next() { int i=0; int j=Next[0]=-1; while(i<len) { while(j!=-1&&s[i]!=s[j]) j=Next[j]; Next[++i]=++j; } } int main() { int t;cin>>t; while(t--) { //memset(num,0,sizeof num); scanf("%d",&len); scanf("%s",s); get_next(); for(int i=0;i<=len;i++) { num[i]=1; int t=Next[i]; while(t>0) { num[t-1]=(num[t-1]+1)%INF; t=Next[t]; } } int ans=0; for(int i=0;i<len;i++) { ans=(ans+num[i])%INF; } printf("%d\n",ans); } return 0; }
hdu3336(Count the string)KMP的应用
时间: 2024-10-05 01:08:00