[Codeforces 919E]Congruence Equation

Description

题库链接

求满足 [ncdot a^nequiv b pmod{p}] 的 (n) 的个数， (1leq nleq x) ， (a,b,p,x) 均已给出。

(2leq pleq 10^6+3,1leq a,b < p, 1leq xleq 10^{12}) ， 保证 (p) 是质数。

Solution

对于 (xleq 10^{12}) 显然不能枚举判断。但我们注意到当关于 (n_1,n_2) 的方程，若满足 (n_1equiv n_2pmod{p(p-1)}) 那么这两个方程就是等价的。

理由可以由费马小定理 (a^{p-1}equiv1pmod{p}) ，以及 (xequiv x-ppmod{p}) 得到。

我们假设 (n=i(p-1)+j) ，那么 [egin{aligned}ncdot a^n&equiv b &pmod{p}\ i(p-1)+j&equiv bcdot a^{-j}&pmod{p}\ j-i&equiv bcdot a^{-j}&pmod{p}end{aligned}]

由于 (j) 可能的取值只有 (p-1) 个，我们可以枚举 (j) 来算出对应的 (i) 的个数，也就是 (n) 的个数。值得注意的是由于 (n) 不能取 (0) 所以为了方便处理，让 (j=0) 变为 (j=p-1) 。

枚举 (j) 后我们可以求出最小的 (i) ： (iequiv j-bcdot a^{-j}pmod{p}) ，进而求出最小的 (n) 。然后求出 ([1,x]) 的范围内的等价的解的个数。

Code

//It is made by Awson on 2018.2.1
#include <bits/stdc++.h>
#define LL long long
#define dob complex<double>
#define lowbit(x) ((x)&(-(x)))
#define Abs(a) ((a) < 0 ? (-(a)) : (a))
#define Max(a, b) ((a) > (b) ? (a) : (b))
#define Min(a, b) ((a) < (b) ? (a) : (b))
#define writeln(x) (write(x), putchar('
'))
#define Swap(a, b) ((a) ^= (b), (b) ^= (a), (a) ^= (b))
using namespace std;
void read(LL &x) {
    char ch; bool flag = 0;
    for (ch = getchar(); !isdigit(ch) && ((flag |= (ch == '-')) || 1); ch = getchar());
    for (x = 0; isdigit(ch); x = (x<<1)+(x<<3)+ch-48, ch = getchar());
    x *= 1-2*flag;
}
void print(LL x) {if (x > 9) print(x/10); putchar(x%10+48); }
void write(LL x) {if (x < 0) putchar('-'); print(Abs(x)); }

LL a, b, p, x;

LL quick_pow(LL a, LL b, LL p) {
    LL ans = 1;
    while (b) {
        if (b&1) ans = ans*a%p;
        a = a*a%p, b >>= 1;
    }
    return ans;
}
void work() {
    read(a), read(b), read(p), read(x);
    LL inva = quick_pow(a, p-2, p);
    LL ans = 0, now = b;
    for (int i = 1; i < p; i++) {
        now = now*inva%p;
        LL n = (p-1)*((i-now+p)%p)+i;
        if (n > x) continue;
        ans += (x-n)/((LL)p*(p-1))+1;
    }
    writeln(ans);
}
int main() {
    work();
    return 0;
}