hdu 3336 Count the string

题目链接http://acm.hdu.edu.cn/showproblem.php?pid=3336

//这题的题意就是给出一个字符串S，求其前缀的子串与该串共有多少个匹配

开始这题理解错题意了，后来明白就简单多了，不过这题有两个优化，要不容易超市

1.对于其子串不必求出所有子串的next数组，只需要求出原串的next，其子串的next则也包含在内

2.当调用KMP时某子串只匹配了一次，那么下面则无需再次调用KMP，解释：如果S是abcabcd,那么当abca只出现一次

那么后面所有前缀子串则只可能匹配一次，例如子串abcab,abcabc,等

代码如下

# include <stdio.h>
# include <string.h>
# define mod 10007
const int NN=200000;
char s[NN+20],p[NN+20];
int next[NN+20],cnt;
int lens,lenp;
void COUNT_next(char *s)
{

    next[0] = 0;
    int k=0,q;
    for(q=1; q<lens; q++)
    {
        while(k>0 && p[k]!=p[q])
            k=p[k];
        if(p[k] == p[q])
            k++;
        next[q]=k;
    }
}
int KMP(char *str,char *p)
{
    int q=0,i;
    cnt=0;
    for(i=0; i<lens; i++)
    {
        while(q>0 && p[q]!=s[i])
            q=p[q];
        if(p[q]==s[i])
        {
            q=q+1;
            if(q==lenp)
            {
                q=0;
                cnt++;
            }
        }
        else q=p[q];
    }
    return cnt;
}
int main()
{
    int t, n, i, q, k;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %s", &n, s);
        int sum=0,k;
        lens=strlen(s);
        COUNT_next(s);
        for(i=0; i<n; i++)
        {
            p[i]=s[i];
            lenp=i+1;
            k=KMP(s,p);
            sum+=k;
            if(k==1)  //这里是第二个优化
            {
                sum=(sum+(n-i-1));
                break;
            }
        }
        printf("%d\n",sum%mod);
    }
    return 0;
}

其实我一开始写的时候也没注意第二个优化，超时好几次

第一篇博文~~~