题面
题解
求 a a a 的 b b b 次方对 p p p 取余,如果仅仅依照乘方的意义进行计算,效率过低,因此,我们可以用到二进制来解决。
根据任何一个十进制数都有唯一的二进制表示可知,任何一个十进制数都可以唯一分解成若干个指数不相同的二的次幂的和。
即 b b b 的二进制表示有 k k k 位,每一位上的数是 c i c_i ci ,则 b = 2 k − 1 × c k − 1 + 2 k − 2 × c k − 2 + . . . + 2 0 × c 0 b = 2^{k-1}\times c_{k-1} + 2^{k-2} \times c_{k-2} +...+ 2^0 \times c_0 b=2k−1×ck−1+2k−2×ck−2+...+20×c0
根据 a n × a m = a n + m a^n \times a^m = a ^ {n + m} an×am=an+m 可知,
a b = a 2 k − 1 × c k − 1 × a 2 k − 2 × c k − 2 × . . . × a 2 0 × c 0 a ^ b = a ^ {2 ^ {k - 1} \times c_{k - 1}} \times a ^ {2 ^ {k - 2} \times c_{k - 2}} \times ... \times a ^{2^0 \times c_0} ab=a2k−1×ck−1×a2k−2×ck−2×...×a20×c0
举一个例子:
2 6 = 2 2 1 × 2 2 2 2 ^ 6 = 2 ^ {2^1} \times 2 ^ {2 ^ 2} 26=221×222
很明显,效率提高了很多。
代码实现(仅展示分解部分):
int power(int a, int b, int p) {
int ans = 1 % p;
for (; b; b >>= 1) {
if (b & 1) ans = (long long)ans * a % p; // b 包含这个二的次幂时答案累乘
a = (long long)a * a % p; // 防止溢出
}
return ans;
}