[CQOI 2018]破解D-H协议

Description

题库链接

给出 (A,B,P,g) ，(g) 是 (P) 的原根，求出 (Aequiv g^apmod{P}) ， (Bequiv g^bpmod{P}) 中的 (a,b) ，并输出 (g^{ab}) 。

(2leq A,B<P<2^{31},2leq g<20,1leq nleq 20)

Solution

(BSGS) 板子啊，由于 (g) 是 (P) 原根，解是唯一的。

Code

#include <bits/stdc++.h>
using namespace std;

map<int, int>mp;
int g, p, n, a, b;

int quick_pow(int a, int b) {
    int ans = 1;
    while (b) {
        if (b&1) ans = 1ll*ans*a%p;
        b >>= 1, a = 1ll*a*a%p;
    }
    return ans;
}
int BSGS(int a, int b) {
    mp.clear();
    int lim = ceil(sqrt(p)), cnt = b%p;
    for (int i = 0; i <= lim; i++, cnt = 1ll*cnt*a%p)
        mp[cnt] = i;
    int t = cnt = quick_pow(a, lim);
    for (int i = 1; i <= lim; i++, cnt = 1ll*cnt*t%p)
        if (mp.count(cnt)) return lim*i-mp[cnt];
}
void work() {
    scanf("%d%d%d", &g, &p, &n);
    while (n--) {
        scanf("%d%d", &a, &b);
        a = BSGS(g, a); printf("%d
", quick_pow(b, a)) ;
    }
}
int main() {work(); return 0; }