hdu3336(kmp)

题目大意:

给定一个字符串s,求s的每个前缀在此字符串中出现的次数,然后对次数求和,然后再对10007取模,就是要输出的答案。

思路:

刚开始也许会想,枚举前缀,求出每个前缀出现的次数,但这样效率太低。

利用kmp算法的next数组可以很好的解决这个问题,    next数组存放的是字符串的前缀和后缀能匹配的字符个数的最大值。

对于i,(1 <= i <=n),n是字符串s的长度

        如果next[i]  == 0,则表示   由s的前i个字符组成的字符串的所有后缀肯定和其前缀不匹配。

        否则      由s的前i个字符组成的字符串存在某个前缀和后缀匹配的情况,也就是该前缀的出现的次数应该加上1。

       定义f[i]为  在由s的前i个字符组成的字符串subs中,subs所有前缀出现的次数之和。答案也就是f[n];

f[i] = f[i-1] + 1 + (next[i]==0?0:1)

在我的程序中没有用f数组,直接用的是ans

hdu3336AC代码:

#include 
#include 

const int maxn = 200005;
char p[maxn];
int n;
int next[maxn];
void getnext(char* p)
{
    next[1] = 0;
    int k = 0;
    for(int i=2;i<=n;i++)
    {
        while(k!=0 && p[k+1] != p[i])
            k = next[k];
        if(p[k+1] == p[i]) k++;
        next[i] = k;
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        scanf("%s",p+1);
        getnext(p);
        int ans = n % 10007;
        for(int i=1;i<=n;i++)
        {
            ans = (ans + (next[i]==0?0:1)) % 10007;
        }
        printf("%d\n",ans);
    }
    return 0;
}


你可能感兴趣的:(kmp)