Sequence

题目描述

给定一个含n个数的序列A和一个含m (m<=n) 个数的序列B。

询问在A中有多少段连续的长为m的子序列Ak,Ak+1,…,Ak+m-1

使得对于任意1<=i, j<=m满足Ak+i-1-Bi=Ak+j-1-Bj

输入

第一行两个整数n，m (1 <= m <= n <= 106)

接下来一行n个整数，描述序列A (Ai <= 109)

接下来一行m个整数，描述序列B (Bi <= 109)

输出

输出一个数表示答案

样例输入

7 4
6 6 8 5 5 7 4
7 7 9 6

样例输出

2

比赛时只想到了求与前一项的差值，然后查找s在r中出现的次数，但超时了，原因是那时不会KMP,现在也是一知半解。不过，先记录下来，慢慢摸索吧!

#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int a[1000020];
int r[1000020];
int b[1000020];
int s[1000020];
int Next[1000020];
void get(int *p,int l)//求Next数组，这一步是关键！！！
{
    int i = 0, j = -1;
    Next[i] = j;
    while(i < l)
    {
        while( j != -1 && s[i] != s[j])
            j =Next[j];
        Next[++i] = ++j;
    }
}
int kmp(int n,int m)//kmp匹配
{
    get(b,m);
    int i = 0, j = 0;
    int ans = 0;
    while(i < n){
        while( j != -1 && s[j] != r[i])
            j = Next[j];
        i++,j++;
        if(j == m-1) ans++;
    }
    return ans;
}
int main()
{
    int n,m,t;
    cin>>n>>m;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
        if(i)r[i-1]=a[i]-a[i-1];//把输入转换为每一项与前一项的差值
    }
    for(int i=0;i<m;i++)
    {
        cin>>b[i];
        if(i)s[i-1]=b[i]-b[i-1];//同上
    }
    cout<<kmp(n,m)<<endl;
    return 0;
}