hdu 6153 A Secret kmp

题意：

给定两个字符串 求一个串每个后缀在另一个串中出现的次数

思路：

把两个串都倒过来 给模式串做kmp在主串上跑一边就行了

#include<bits/stdc++.h>
#define cl(a,b) memset(a,b,sizeof(a))
#define debug(a) cerr<<#a<<"=="<<a<<endl
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;

const int maxn=1e6+10;
const int mod=1e9+7;

int _next[maxn],val[maxn];
char a[maxn],b[maxn];

inline void init()
{
    cl(_next,0),cl(val,0);
}

void kmp_pre(char x[],int _next[],int val[])
{
    int i,j;
    _next[0]=-1,_next[1]=0;
    val[0]=0,val[1]=1;
    int len=strlen(x);
    for(i=1; i<len; i++)
    {
        _next[i+1]=0;
        val[i+1]=i+1;
        j=_next[i];
        while(j>=0)
        {
            if(x[i]==x[j])
            {
                _next[i+1]=j+1;
                val[i+1]+=val[j+1];
                val[i+1]%=mod;
                break;
            }
            j=_next[j];
        }
    }
}


ll KMP_Count(char x[],int _next[],char y[],int val[])
{
    int len=strlen(y),i,j,k;
    ll ans=0;
    i=j=k=0;
    kmp_pre(x,_next,val);
    for(; i<len; i++)
    {
        k=j;
        j=0;
        while(k>=0)
        {
            if(y[i]==x[k])
            {
                j=k+1;
                break;
            }
            k=_next[k];
        }
        ans+=val[j];
        ans%=mod;
    }
    return ans;
}

void get()
{
    int lena=strlen(a);
    int lenb=strlen(b);
    for(int i=0; i<lena/2; i++)
    {
        swap(a[i], a[lena-i-1]);
    }
    for(int i=0; i<lenb/2; i++)
    {
        swap(b[i], b[lenb-i-1]);
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%s",a);
        scanf("%s",b);
        get();
        printf("%lld
",KMP_Count(b,_next,a,val));
    }
    return 0;
}/*

2
aaaaa
aa
abababab
aba

*/