1. 首页
2. KMP
3. 本页

HDU 5918 Sequence I (KMP)

题目链接：点我~~

题意：给定序列a、序列b和一个整数$p$，要求出有多少个$q$使得b1,b2,…,bm恰好是aq,aq+p,…,aq+(m−1)p。

思路：将a按照$\mathrm{MOD}p$拆成若干个串，分别做一次 KMP.

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PI;
typedef pair< PI, int> PII;
const double eps=1e-5;
const double pi=acos(-1.0);
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
#define Key_value ch[ch[root][1]][0]
const int MAXN = 100000+100;
const int MAXM = 10100;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1

void kmp_pre(int x[],int m,int nex[])
{
    int i,j;
    j=nex[0]=-1;
    i=0;
    while(i<m)
    {
        while(-1!=j && x[i]!=x[j])j=nex[j];
        nex[++i]=++j;
    }
}

int nex[1000010];
int p,q;
int KMP_Count(int x[],int m,int y[],int n)
{
    int i,j;
    int ans=0;
    kmp_pre(x,m,nex);
    j=0;
    i=q;
    while(i<n)
    {
        while(-1!=j && y[i]!=x[j])
            j=nex[j];
        i+=p;
        j++;
        if(j>=m)
        {
            ans++;
            j=nex[j];
        }
    }
    return ans;
}

int a[1000000+10];
int b[1000000+10];

int main()
{
    int t;
    scanf("%d",&t);
    int casee=1;
    while(t--)
    {
        int n,m;
        scanf("%d%d%d",&n,&m,&p);
        for(int i=0; i<n; ++i) scanf("%d",&a[i]);
        for(int i=0; i<m; ++i) scanf("%d",&b[i]);
        printf("Case #%d: ",casee++);
        int ans=0;
        int len=min(p,n-(m-1)*p);
        for(int i=0;i<len;++i)
        {
            q=i;
            ans+=KMP_Count(b,m,a,n);
        }
        cout<<ans<<endl;
    }

    return 0;
}