hdu5918Sequence I

链接：http://acm.hdu.edu.cn/showproblem.php?pid=5918

题意：给出两个数组a,b。b数组匹配a数组，不是连续匹配，而是相隔p位置匹配，求a数组中能匹配出多少次b数组。

分析：将a数组按%p分组然后kmp即可，O(n+m)。

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=1e6+10;
const int M=5e4+10;
const int mod=1000000007;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000007;
const int INF=1000000010;
const ll MAX=1ll<<55;
const double eps=1e-5;
const double inf=~0u>>1;
const double pi=acos(-1.0);
typedef double db;
typedef unsigned int uint;
typedef unsigned long long ull;
int a[N],b[N],c[N],nex[N];
void pre_kmp(int s[],int len) {
    int i,k;
    nex[0]=k=-1;
    for (i=0;i<len;i++) {
        while (k>-1&&s[k]!=s[i]) k=nex[k];
        k++;nex[i+1]=k;
    }
}
int kmp(int s[],int les,int len) {
    int i,k=0,ret=0;
    for (i=0;i<len;i++) {
        while (k>0&&b[k]!=s[i]) k=nex[k];
        if (b[k]==s[i]) k++;
        if (k==les) ret++,k=nex[k];
    }
    return ret;
}
int main()
{
    int i,j,n,m,q,ca,T,ans;
    scanf("%d", &T);
    for (ca=1;ca<=T;ca++) {
        scanf("%d%d%d", &n, &m, &q);
        for (i=0;i<n;i++) scanf("%d", &a[i]);
        for (i=0;i<m;i++) scanf("%d", &b[i]);
        ans=0;
        pre_kmp(b,m);
        for (i=0;i<q&&i<n;i++) {
            for (j=0;i+j*q<n;j++) c[j]=a[i+j*q];
            ans+=kmp(c,m,j);
        }
        printf("Case #%d: %d\n", ca, ans);
    }
    return 0;
}