Codeforces Round #269 (Div. 2) D.MUH and Cube Walls

【题目】点击打开链接

【题意】其实就是处理两个数组的差值，然后去看有多少个匹配。

【解题方法】KMP

【AC代码】

#include <bits/stdc++.h>
using namespace std;
const int maxn=200010;
int n,m;
int a[maxn],b[maxn];
int h[maxn],w[maxn];
int nex[maxn];
int ans;
void getfail()
{
    nex[0]=nex[1]=0;
    for(int i=1;i<m;i++){
        int j=nex[i];
        while(j!=0&&b[j]!=b[i]) j=nex[j];
        nex[i+1]=(b[i]==b[j])?j+1:0;
    }
}
void kmp()
{
    int j=0;
    for(int i=0;i<n;i++){
    while(j!=0&&b[j]!=a[i]) j=nex[j];
    if(b[j]==a[i]) j++;
    if(j==m){
            ans++;
            j=nex[j];
        }
    }
}
int main()
{
    cin>>n>>m;
    for(int i=0; i<n; i++) cin>>h[i];
    for(int i=0; i<m; i++) cin>>w[i];
    for(int i=0; i<n; i++){
        if(i>=1){
            a[i-1]=h[i]-h[i-1];
        }
    }
    for(int i=0; i<m; i++){
        if(i>=1){
            b[i-1]=w[i]-w[i-1];
        }
    }
    n--,m--;
    if(m==0) cout<<n+1<<endl;
    else{
        ans=0;
        getfail();
        kmp();
        cout<<ans<<endl;
    }
}