A Secret HDU

A Secret

 HDU - 6153

 题意：给两个串s1,s2,问s2的所有后缀在s1中出现的次数乘以后缀长度的和是多少。

首先将串翻转，然后KMP即可。

通过这个题我也认识到对很多简单的算法理解的不够深刻，学了也就只会耍耍模板，以后要多思考思考。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e6+10;
const int mod=1e9+7;
char s[maxn],p[maxn];
int f[maxn];
int lens,lenp;
ll ans;

void getfail(char* p){
    f[0]=f[1]=0;
    for(int i=1;i<lenp;i++){
        int j=f[i];
        while(j&&p[i]!=p[j]) j=f[j];
        f[i+1]=p[i]==p[j]?j+1:0;
    }
    return ;
}

void KMP(char* s,char* p){
    getfail(p);
    ll j=0;
    for(ll i=0;i<lens;i++){
        while(j&&s[i]!=p[j]) {
            ans=(ans+(j+1)*j/2)%mod;
            j=f[j];
        }
        if(s[i]==p[j]) j++;
        if(j==lenp){
            ans=(ans+(j+1)*j/2)%mod;
            j=f[j];
        }
    }
    while(j){
      ans=(ans+j*(j+1)/2)%mod;
      j=f[j];
    }
    return ;
}

int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        ans=0;
        scanf("%s%s",s,p);
        lens=strlen(s);
        lenp=strlen(p);
        reverse(s,s+lens);
        reverse(p,p+lenp);
        KMP(s,p);
        printf("%lld
",ans);
    }
    return 0;
}

上面的代码虽然过了，可是对于这组数据答案是错的=_=

baabaa

caabaa

23

看到了扩展KMP~去学习下再来补~

是可以用扩展KMP秒掉的~

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e6+10;
const int mod=1e9+7;

char s[maxn],t[maxn];
int nex[maxn],ex[maxn];
int lens,lent;
void getnex(char* t){
    nex[0]=lent;
    int a=0,p=0;
    for(int i=1;i<lent;i++){
        if(i>=p||i+nex[i-a]>=p){
            if(i>=p) p=i;
            while(p<lent&&t[p]==t[p-i]) p++;
            nex[i]=p-i;
            a=i;
        }else nex[i]=nex[i-a];
    }
}
void exkmp(char* s,char* t){
    getnex(t);
    int a=0,p=0;
    for(int i=0;i<lens;i++){
        if(i>=p||i+nex[i-a]>=p){
            if(i>=p) p=i;
            while(p<lens&&p-i<lent&&s[p]==t[p-i]) p++;
            ex[i]=p-i;
            a=i;
        }else ex[i]=nex[i-a];
    }
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%s%s",s,t);
        lens=strlen(s);
        lent=strlen(t);
        reverse(s,s+lens);
        reverse(t,t+lent);
        exkmp(s,t);
        ll ans=0;
        for(int i=0;i<lens;i++) if(ex[i]) ans=(ans+1ll*(ex[i]+1)*ex[i]/2)%mod;
        printf("%lld
",ans);
    }
}