[CodeForces-471D] kmp

题意:
给定两个序列
在a中找到一个子串 使得每个数同时加上某个整数(可以为负数)后和 b 相等
问有多少个这样的子串
思路:
我们从变量中寻找不变量,对两个序列作前缀差,这样就把变量变成了不变量,然后跑kmp模板,巧妙!

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=2e5+5;
int Next[maxn];
void getnext(int *ptr,int n)
{
    int i,k;
    k=-1;
    Next[0]=-1;
    i=0;
    while(i<n)
    {
        if(k==-1 || ptr[i]==ptr[k])
        {
            i++;
            k++;
            Next[i]=k;
        }
        else k=Next[k];
    }
}
int kmp(int *a,int *b,int len1,int len2)//查找子串出现个数
{
    int i=0,j=0,ans=0;
    getnext(b,len2);
    while(i<len1)
    {
      if(j==-1||a[i]==b[j])
        ++i,++j;
      else
        j=Next[j];
      if(j==len2)//匹配到一个了就当没找到然后跳到next[J] 
		{
			j=Next[j];
			ans++;
		}
    }
    return ans;
}
int s[maxn],p[maxn];
int main()
{
	int n,m;
    scanf("%d%d",&n,&m);
    for(int i=0; i<n; i++)
    {
        scanf("%d",&s[i]);
        if(i)s[i-1]=s[i]-s[i-1];
    }
    for(int i=0; i<m; i++)
    {
        scanf("%d",&p[i]);
        if(i)p[i-1]=p[i]-p[i-1];
    }
    if(m==1)
    {
    	printf("%d",n);
    	return 0;
	}
    printf("%d\n",kmp(s,p,n-1,m-1));
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值