【Codeforces】Round #460 E - Congruence Equation 中国剩余定理+数论

题意

求满足$na^n\equiv b \pmod p$的$n$的个数

因为$n \mod p ​$循环节为$p​$，$a^n\mod p​$循环节为$p-1​$，所以$na^n \mod p​$循环节为$p(p-1)​$

假设$n \mod p \equiv i,a^n\mod p\equiv a^j$ ， 那么$n%p \times a^n%p\equiv b \pmod p$，得到$i \times a^j \equiv b \pmod p$，列同余方程$i \equiv b*a^{-j} \pmod p, i\equiv j \pmod {p-1}$，解得$i=(p-1)^2ba^j+pj$，在$n$的上限内计算答案

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
LL a, b, p, x, ans = ;
LL quick_pow(LL x, LL y, LL mod) {
    LL ret = ;
    for(; y; y >>= ) {
        if(y & ) ret = (ret * x) % mod;
        x = (x * x) % mod;
    }
    return ret;
}
int main() {
    cin >> a >> b >> p >> x;
    for(LL i = ; i < p; ++i) {
        LL inv = quick_pow(quick_pow(a, i, p), p - , p);
        LL y = b * inv % p;
        LL P = p * (p - );
        LL r = (p * i + (p - ) * (p - ) % P * y) % P;
        ans += x / P + (x % P >= r);
    }
    cout << ans << endl;
    return ;
}