HUST 1328 String

11:

    KMP next 的强大

  题意求前缀在S中出现的次数之和

next[j] 表示 S[0....NEXT[J]]==S[J-NEXT[J].....J];

于是我们得到。。后加入一个字符所得到新的前缀会多ADD[next[J]]个

 

#include<stdio.h>

 

#include<string.h>
#include<algorithm>
#include<iostream>
#include<math.h>
using namespace std;
#define mod 1000000007
typedef long long ll;
#define N 123456
char s[N];
int next[N];
int n;
ll a[N];
void kmp()
{
    n=strlen(s);
    int i=0,j=-1;
    next[0]=-1;
    while (i<n)
    {
        if (j==-1||s[i]==s[j])
        {
            i++;
            j++;
            next[i]=j;
        }else j=next[j];
    }
}
int main()
{
    while (scanf("%s",s)!=EOF)
    {
        kmp();
        ll ans=0;
        memset(a,0,sizeof(a));
        for (int i=1;i<=n;i++)
        {
            a[i]=a[next[i]];
            a[i]++;
            ans+=a[i];
        }
        printf("%lld\n",ans);
    }
    return 0;
}

posted on 2015-02-12 01:46  forgot93  阅读(143)  评论(0编辑  收藏  举报

导航