1. 程式人生 > >kmp next 陣列的巧妙運用+dp

kmp next 陣列的巧妙運用+dp

題意:

        給你一個串,求用該串所有字首去 匹配本身這個串的次數 的總和。比如串abab,它的字首有a,ab,aba,abab。那麼拿這4個字首去匹配abab自身分別有2,2,1,1個匹配點,所以總和為2+2+1+1=6。

思想:

其實我們想想比如對於位置i來說,現在我們求出了f[i]的值,那麼[0,f[i]-1]串就是一個能匹配i-1位置的最長的字首串。且我們根據f[i]的值可以定位其他所有可匹配的字首。KMP的思想就是找出每個位置i的可匹配最大字首j。

        令dp[i]表示S[0,i-1]串的字尾能匹配的字首個數,那麼dp[i]=dp[next[i]]+1, dp[1]=1正好表示串s[0]的字尾只能匹配串s[0].
 

<span style="font-size:18px;">#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int MAXM=200000+100;
const int MOD=10007;
char P[MAXM];
int f[MAXM],dp[MAXM];
int m;
void getFail(char *P,int *f)
{
    f[0]=f[1]=0;
    for(int i=1;i<m;i++)
    {
        int j=f[i];
        while(j && P[i]!=P[j]) j=f[j];
        f[i+1] = (P[i]==P[j])? j+1:0;
    }
}
int main()
{
    int k;
    scanf("%d",&k);
    while(k--)
    {
        scanf("%d %s",&m,P);
        getFail(P,f);
        dp[0]=0;
        int ans=0;
        for(int i=1;i<=m;i++)
        {
            dp[i]=dp[f[i]]+1;
            dp[i]%=MOD;
            ans=(ans+dp[i])%MOD;
        }
        printf("%d\n",ans);
    }
    return 0;
}
</span>