Count the string HDU - 3336

http://acm.hdu.edu.cn/showproblem.php?pid=3336

问每个前缀可重叠地出现了多少次

 

KMP解法

最笨的方法就是对于每个i 不断找next[i] 看找几次到0 累加一下 即看以i为结尾的字符串中有多少和整个串的某个前缀相等

但是aaaaaa直接是n^2 复杂度爆炸 这个过程中很多步都是重复走的 可以用和差分数组类似的方法 在找的过程中不断累加

 

 

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const ll mod=1e4+7;
const int maxn=2e6+10;

ll book[maxn];
int nxtt[maxn];
char ch[maxn];

void kmp(char *t,int lt)
{
    int i,j;
    nxtt[0]=-1;
    i=0,j=-1;
    while(i<lt)
    {
        if(j==-1||t[i]==t[j])
        {
            i++,j++;
            nxtt[i]=j;
        }
        else j=nxtt[j];
    }
}

int main()
{
    ll ans;
    int t,l,i;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%s",&l,ch);
        kmp(ch,l);
        memset(book,0,sizeof(book));
        for(i=l;i>=1;i--){
            book[nxtt[i]]=(book[nxtt[i]]+book[i]+1)%mod;
        }
        ans=l;
        for(i=1;i<=l;i++) ans=(ans+book[i])%mod;
        printf("%lld\n",ans);
    }
    return 0;
}

 

EXKMP解法

思路相较KMP简单一些 跑一遍模板 将nxtt数组累加即可 因为nxtt[i]在这就代表 以i为首字母能找出多少前缀

 

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int mod=1e4+7;
const int maxn=2e5+10;

int nxtt[maxn];
char ch[maxn];
int n;

void exkmp(char *t,int lt)
{
    int i,j,p;
    nxtt[0]=lt;
    i=0;
    while(i+1<lt&t[i]==t[i+1]) i++;
    nxtt[1]=i;
    p=1;
    for(i=2;i<lt;i++){
        if(i+nxtt[i-p]<p+nxtt[p]) nxtt[i]=nxtt[i-p];
        else{
            j=max(0,p+nxtt[p]-i);
            while(i+j<lt&&t[j]==t[i+j]) j++;
            nxtt[i]=j;
            p=i;
        }
    }
}

int main()
{
    int t,i,ans;
    scanf("%d",&t);
    while(t--){
        scanf("%d%s",&n,ch);
        exkmp(ch,n);
        ans=0;
        for(i=0;i<n;i++){
            ans=(ans+nxtt[i])%mod;
        }
        printf("%d\n",ans);
    }
    return 0;
}