在这道题中首先可以发现因为是在mod意义下的乘方,那么可以看做是mod p之后的a数列进行乘方,那么就将每一个a数列中的数控制在了0-(p-1)之间因为mod是一个质数那么
ap−1modp=1modp
那么就可以将b数列控制在0-(p-2)之间,那么只要统计a数列中出现的每一个次数,b出现的每一个次数,然后暴力算就好了复杂度
O(p2)
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int MAXN = 5000;
int cta[MAXN+10], ctb[MAXN+10];
int main(){
int p, q, n, m, A, B, C, D, E, F, a1, a2, b1, b2;
scanf("%d%d", &p, &q);
scanf("%d%d%d%d%d%d", &n, &a1, &a2, &A, &B, &C);
scanf("%d%d%d%d%d%d", &m, &b1, &b2, &D, &E, &F);
cta[((a1%p)+p)%p] ++;
cta[((a2%p)+p)%p] ++;
ctb[((b1%(p-1))+(p-1))%(p-1)] ++;
ctb[((b2%(p-1))+(p-1))%(p-1)] ++;
A%=p; B%=p; C%=p; D%=(p-1); E%=(p-1); F%=(p-1);
for(int i=3;i<=n;i++){
int t = ((A*a2-B*a1-C)%p+p)%p;
cta[t]++;
a1=a2; a2=t;
}
int tp = p-1;
for(int i=3;i<=m;i++){
int t = ((D*b2+E*b1+F)%tp+tp)%tp;
ctb[t]++;
b1=b2; b2=t;
}
long long ans = 0;
for(int i=0;i<p;i++){
if(!cta[i]) continue;
int now = 1;
for(int j=0;j<tp;j++){
if(now <= q)
ans += 1LL * cta[i] * ctb[j];
now = now * i % p;
}
}
cout<<ans<<endl;
return 0;
}