您的位置:首页 > 其它

hdu3336(Count the string)KMP的应用

2014-05-08 23:30 141 查看
题意:给一个字符串,计算所有前缀在字符串中出现的次数和。

解法: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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: