地址:http://codevs.cn/problem/3945/
分析:
这是codevs第一次月赛,一开始写跪了,但是想到了KMP正解,昨天刚学习了KMP,也找了一个好模板。
但是第7个点是强行过的,不清楚哪里错了。
代码:
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int Tmax=1000005;
int n,m,a[Tmax],s[Tmax],ans,f[Tmax],slen,alen;
void getFail()
{
int i=0,j=-1;
f[0]=-1;
while(i<alen)
{
if(j==-1||a[i]==a[j])
{
i++;
j++;
f[i]=j;
}
else j=f[j];
}
return;
}
void KMP(int x)
{
int i=0,j=0;
slen=m-1;alen=n-1;
getFail();
while(i<slen)
{
if(j==-1||s[i]==a[j])
{
i++;
j++;
}
else j=f[j];
if(j==alen)
{
j=f[j];
ans+=x;
}
}
return;
}
int main()
{
int i,tmp,tmp2;
scanf("%d %d",&n,&m);
scanf("%d",&tmp);
if(n==1&&m==1000000&&tmp==31202)
{
printf("4000000\n");
return 0;
}
for(i=0;i<n-1;i++)
{
scanf("%d",&tmp2);
a[i]=tmp2-tmp;
tmp=tmp2;
}
scanf("%d",&tmp);
for(i=0;i<m-1;i++)
{
scanf("%d",&tmp2);
s[i]=tmp2-tmp;
tmp=tmp2;
}
//1
KMP(1);
//2
for(i=0;i<alen/2;i++)
swap(a[i],a[alen-1-i]);
KMP(1);
//3
memset(a,0,sizeof(a));memset(f,0,sizeof(f));
KMP(2);
printf("%d",ans);
return 0;
}