题目描述
栗酱有一个长度为n的数列A,一个长度为m的数列B,现在询问A中有多少个长度为m的连续子序列A',
满足(a' 1+b 1)%k = (a' 2+b 2)%k = …… = (a' m + b m)%k。
满足(a' 1+b 1)%k = (a' 2+b 2)%k = …… = (a' m + b m)%k。
输入描述:
第一行一个数T,表示有T组数据。对于每组数据,第一行三个整数,n, m, k。第一行输入n个数, a 1,a 2,…,a n, 表示A数列中的数,第二行输入m个数, b 1,b 2,…,b m, 表示B数列中的数。
输出描述:
每一组数据输出一行,满足条件的连续子序列数量。
示例1
输入
2 3 2 5 7 8 7 8 7 3 2 5 7 8 9 8 7
输出
1 2
备注:
T≤15,2≤m≤n≤2×10 5,1≤a i,b i,k≤10 9
由(a'1+b1)%k = (a'2+b2)%k = …… = (a'm + bm)%k转换一下
(a1+b1)%k=(a2+b2)%k
(a2+b2−a1−b1)%k=0
即
((a2−a1)+(b2−b1))%k=0
那么只需要存一下da[i]=a[i+1]-a[i]和db[i]=b[i+1]-a[i]
修改一下KMP的条件,让da匹配db即可
过程:对db求next数组,用da匹配db,把匹配时的判断改为(da[i]+db[i])%k==0
#include<bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const int N=2e5+50;
int nxt[N],a[N],b[N];
int da[N],db[N];
int n,m,k,t;
void get_nxt()
{
int i=0,j=-1;
nxt[0]=-1;
while(i<m)
{
if(j==-1||db[i]==db[j])
{
nxt[++i]=++j;
}
else
j=nxt[j];
}
}
void kmp()
{
get_nxt();
int ans=0,i=0,j=0,la=n,lb=m;
while(i<la)
{
if(j==-1||(da[i]+db[j])%k==0)
{
++i,++j;
}
else
j=nxt[j];
if(j==lb)ans++,j=nxt[j];
}
printf("%d\n",ans);
}
int main()
{
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&m,&k);
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
a[i]%=k;
}
for(int i=0;i<m;i++)
{
scanf("%d",&b[i]);
b[i]%=k;
}
for(int i=0;i<n-1;i++)
da[i]=(a[i+1]-a[i]+k)%k;
for(int i=0;i<m-1;i++)
db[i]=(b[i+1]-b[i]+k)%k;
n--;
m--;
kmp();
}
return 0;
}