给定一个题目,求a^b mod p,如果用暴力(循环)去求可能会导致超时,我们可以考虑使用快速幂算法,可以很快的求出结果。算法其实来自一句初中数学的定理,同底数幂相乘,底数不变,指数相加。所以我们可以将b写成几个数相加的形式,但要怎么写呢,没错,用二进制。比如11的二进制可以写成1011(8+2+1),所以
a
11
a^{11}
a11可以写成
a
8
∗
a^{8}*
a8∗
a
2
∗
a^{2}*
a2∗
a
1
a^{1}
a1,用二进制还有一个好处就是方便使用位运算。每次判断b的最低位是否为1,是则要乘以一个a的幂,否则不需要,然后将b右移一位,同时a要乘以a,因为此时已经到了b的下一位。下面来看题目和相应的代码。
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
int a, b, p;
cin >> a >> b >> p;
int rel = 1 % p;
while(b)
{
if(b & 1) rel = (long long) rel * a % p;
b >>= 1;
a = (long long)a * a % p;
}
cout << rel << endl;
return 0;
}