[AcWing] 快速幂求逆元

快速幂求逆元

题目

给定n组ai,pi，其中pi是质数,求ai模pi的乘法逆元，若逆元不存在则输出impossible。

注意：请返回在0∼p−1之间的逆元。

乘法逆元的定义

若整数b，m互质，并且对于任意的整数 a，如果满足b|a，则存在一个整数x，使得a/b≡a∗x(mod
m)，则称x为b的模m乘法逆元，记为b−1(mod m)。
b存在乘法逆元的充要条件是b与模数m互质。当模数m为质数时，bm−2即为b的乘法逆元。

输入格式

第一行包含整数n。

接下来n行，每行包含一个数组ai,pi，数据保证pi是质数。

输出格式

输出共n行，每组数据输出一个结果，每个结果占一行。

若ai模pi的乘法逆元存在，则输出一个整数，表示逆元，否则输出impossible。

数据范围

$$\begin{gathered} 1\le n\le 10^5 \\ 1\le ai,pi\le 2\ast 10^9 \end{gathered}$$

输入样例

3
4  3
8  5
6  3

输出样例

1
2
impossible

题解

思路

• 由费马定理可知：如果 a 与 p 互质，那么，$a^{p-1}\equiv 1\mathrm{m}\mathrm{o}\mathrm{d}p,a\ast a^{p-2}\equiv 1\mathrm{m}\mathrm{o}\mathrm{d}p$
• 则 a 的逆元为 $a^{p-2}\equiv 1\mathrm{m}\mathrm{o}\mathrm{d}p$

注意

• 如果 a 与 p 不互质，那么，a 没有逆元

#include <iostream>

using namespace std;

typedef unsigned long long LL;

LL qmi(LL a, LL k, LL p) {
    LL res = 1;
    while (k) {
        if (k & 1) res = res * a % p;
        a = a * a % p;  
        k >>= 1;
    }
    return res % p;
}

int main () {
    int t;
    LL a, b;
    cin >> t;
    while (t --) {
        cin >> a >> b;
        LL res = qmi(a, b - 2, b);
        if (a % b) cout << res << endl;
        else puts("impossible");
    }
    
    return 0;
}