题目描述
求 a 的 b 次方对 p 取模的值。
输入格式
三个整数 a、b、p ,在同一行用空格隔开。
输出格式
输出一个整数,表示 a^b mod p
的值。
数据范围
1 ≤ a , b , p ≤ 1 0 9 1 ≤ a , b , p ≤ 10^9 1≤a,b,p≤109
输入样例:
3 2 7
输出样例:
2
解题思路
本题目就是一道快速幂乘的题目。
这里再次解释一下快速幂乘的原理。
假定求解 a b a ^b ab 的值,并且 b ⇔ k n k n − 1 … … k 0 ( k i ∈ { 0 , 1 } ) b \Leftrightarrow k_nk_{n-1}……k_0 (k_i \in \{0,1\}) b⇔knkn−1……k0(ki∈{0,1}) 。
可以得到如下推论: a b ⇔ a k n k n − 1 … … k 0 ⇔ a 2 n × k n + 2 n − 1 × k n − 1 + … … + 2 0 × k 0 ⇔ a 2 n × k n × a 2 n − 1 × k n − 1 × … … × a 2 0 × k 0 a^b \Leftrightarrow a^{k_nk_{n-1}……k_0} \Leftrightarrow a^{2^{n} \times k_n + 2^{n-1} \times k_{n-1} + …… + 2^0 \times k_0} \Leftrightarrow a^{2^n \times k_n} \times a^{2^{n-1} \times k_{n-1}} \times …… \times a^{2^0 \times k_0} ab⇔aknkn−1……k0⇔a2n×kn+2n−1×kn−1+……+20×k0⇔a2n×kn×a2n−1×kn−1×……×a20×k0
对于任意 a 2 n × k n a^{2^n \times k_n} a2n×kn,如果 k n = = 0 k_n == 0 kn==0 ,则 a 2 n × k n = 1 a^{2^n \times k_n} = 1 a2n×kn=1 。如果 k n = = 1 k_n == 1 kn==1 ,则 a 2 n × k n = a 2 n − 1 × a 2 n − 1 a^{2^n \times k_n} = a^{2^{n-1}} \times a^{2^{n-1}} a2n×kn=a2n−1×a2n−1 。
我们可以依次记录所有的 a 2 i a^{2^i} a2i ,然后根据对应的 k i k_i ki 得到 a b a^b ab 的值。
本题目唯一需要注意一点的就是数值溢出。
#include <iostream>
using namespace std;
int main()
{
long a,b,p;
cin >> a >> b >> p;
long result = 1 % p,base = a;
while(b)
{
if(b & 1)
{
result = (result * base) % p;
}
b = b >> 1;
base = (base * base) % p;
}
cout << result << endl;
return 0;
}