zzuli-2264 sequence

[题目链接]

思路：裸裸的kmp，只是一直没敲过，恨自己没带模板~~
代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int Max_n=1e6+10;

int n,m;
int a[Max_n],b[Max_n];
int next1[Max_n];

void getnext1(){
    int i=0,j=-1;
    next1[0]=-1;
    while(i<n-1){
        if(j==-1||b[i]==b[j]){
            i++;j++;
            next1[i]=j;
        }
        else j=next1[j];
    }
}

int kmp_count(){
    int i=0,j=0,k=0;
    getnext1();
    while(i<n-1){
        if(j==-1||a[i]==b[j]){
            i++;j++;
        }
        else next1[j]=j;
        if(j==m-1){
            k++;
            j=next1[j];
        }
    }
    return k;
}

int main()
{
    while(~scanf("%d%d",&n,&m)){
        for(int i=0;i<n;i++)scanf("%d",&a[i]);
        for(int i=0;i<m;i++)scanf("%d",&b[i]);
        if(m==1){
            printf("%d\n",n);
            continue;
        }
        for(int i=0;i<n-1;i++)a[i]-=a[i+1];
        for(int i=0;i<m-1;i++)b[i]-=b[i+1];
        printf("%d\n",kmp_count());
    }
    return 0;
}