用于快速求
a
b
m
o
d
p
a^b mod \ p
abmod p的结果,时间复杂度是
O
(
l
o
g
k
)
O(logk)
O(logk)
其中
1
≤
a
,
b
,
p
≤
1
0
9
1 \leq a,b,p \leq 10^9
1≤a,b,p≤109
核心思想:
-
预处理 a 2 b m o d p a^{2^b} mod \ p a2bmod p的结果, b = 0 , 1 , 2... b=0,1,2... b=0,1,2...
-
对于任意求的 a k m o d p a^k mod \ p akmod p的结果,将其拆成 a 2 x 1 + 2 x 2 + . . . = a 2 x 1 ∗ a 2 x 2 ∗ . . . a^{2^{x_1} + 2^{x_2} + ...}=a^{2^{x_1}} * a^{2^{x_2}} * ... a2x1+2x2+...=a2x1∗a2x2∗...
例如预处理了 a 1 , a 2 , a 4 , a 8 a^1,a^2,a^4,a^8 a1,a2,a4,a8,则 a 14 = a 8 ∗ a 4 ∗ a 2 a^{14}=a^8*a^4*a^2 a14=a8∗a4∗a2, 14 14 14的二进制是 1110 1110 1110,在1的位置乘上 a x i a^{x_i} axi。 -
预处理只需要对 a 2 b a^{2^b} a2b平方,就可以得到 a 2 b + 1 a^{2^{b+1}} a2b+1,迭代 b − 1 b-1 b−1次即可。
如 a 1 a^1 a1平方得到 a 2 a^2 a2, a 2 a^2 a2平方得到 a 4 a^4 a4,依次类推
#include <iostream>
using namespace std;
int n;
int quick_pow(int a, int b, int p) {
int ans = 1;
while (b) {
if (b & 1) ans = (long long)ans * a % p; // 转化成二进制,1的位置乘上a对应次幂
b >>= 1;
a = (long long)a * a % p; // 对a^(2^b)进行平方
}
return ans;
}
int main() {
int a, b, p;
scanf("%d", &n);
while (n -- ) {
scanf("%d%d%d", &a, &b, &p);
printf("%d\n", quick_pow(a, b, p));
}
return 0;
}