KMP-MUH and Cube Walls-CodeForces - 471D

KMP-MUH and Cube Walls-CodeForces - 471D

题意:

$$\begin{gathered} 给定长度为n的整数序列a和长度为m的整数序列b，其中b序列可以将所有数同时增加或减少h， \\ 问在可以修改b的情况下，序列b在a中最多能够匹配成功几次。 \end{gathered}$$

输入输出：

$$\begin{gathered} 如样例：将序列b同时增加1得到：45543，与序列a第二个位置起的子串成功匹配。 \\ 将序列b同时减少1得到：23321，与序列a的长度为5的后缀成功匹配。 \\ 因此共成功匹配两次。 \end{gathered}$$

题解：

$序列匹配问题，考虑KMP算法。$

$$\begin{gathered} 事实上，无论序列b同时增加或减少多少，相邻两个数的差是不会改变的。 \\ 因此，我们只需要将序列a和b差分一次，再去匹配它们的差分数组即可。 \end{gathered}$$

注意：

$差分数组的长度比原数组的长度小1。$

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<map>
#define ll long long
#define inf 0x7fffffff
using namespace std;
const int N=2e5+10;
int Next[N],n,m;
int a[N],b[N];
int da[N],db[N];
void get_next()
{
    for(int i=2,j=0;i<m;i++)
    {
        while(j&&db[i]!=db[j+1]) j=Next[j];
        if(db[i]==db[j+1]) j++;
        Next[i]=j;
    }
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=m;i++) scanf("%d",&b[i]);

    if(m==1) {cout<<n<<endl;return 0;}

    for(int i=1;i<n;i++) da[i]=a[i+1]-a[i];
    for(int i=1;i<m;i++) db[i]=b[i+1]-b[i];

    get_next();

    int cnt=0;
    for(int i=1,j=0;i<n;i++)
    {
        while(j&&da[i]!=db[j+1]) j=Next[j];
        if(da[i]==db[j+1]) j++;

        if(j==m-1)
        {
            cnt++;
            j=Next[j];
        }
    }

    printf("%d\n",cnt);
    return 0;
}