【数论】[NOIP模拟赛]无聊的计算

在这道题中首先可以发现因为是在mod意义下的乘方，那么可以看做是mod p之后的a数列进行乘方，那么就将每一个a数列中的数控制在了0-(p-1)之间因为mod是一个质数那么$a^{p-1}\mod p=1\mod p$那么就可以将b数列控制在0-(p-2)之间，那么只要统计a数列中出现的每一个次数，b出现的每一个次数，然后暴力算就好了复杂度$O(p^2)$

#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;
}