题目地址:
https://www.luogu.com.cn/problem/P1226
题目描述:
给你三个整数
a
,
b
,
p
a,b,p
a,b,p,求
a
b
m
o
d
p
a^b \bmod p
abmodp。
输入格式:
输入只有一行三个整数,分别代表
a
,
b
,
p
a,b,p
a,b,p。
输出格式:
输出一行一个字符串a^b mod p=s
,其中
a
,
b
,
p
a,b,p
a,b,p分别为题目给定的值,
s
s
s为运算结果。
数据范围:
对于
100
%
100\%
100%的数据,保证
0
≤
a
,
b
<
2
31
0\le a,b < 2^{31}
0≤a,b<231,
a
+
b
>
0
a+b>0
a+b>0,
2
≤
p
<
2
31
2 \leq p \lt 2^{31}
2≤p<231。
思路是快速幂(或者也可以称为叫“倍增法”),由 a a a出发,依次计算 a 2 n , n = 0 , 1 , 2 , . . . a^{2^n},n=0,1,2,... a2n,n=0,1,2,...,然后将 b b b看成二进制,倍增地计算 a b m o d p a^b\mod p abmodp。代码如下:
#include <iostream>
using namespace std;
int a, b, p;
int fast_pow(int a, int b, int p) {
long res = 1 % p, t = a;
while (b) {
if (b & 1) res = res * t % p;
b >>= 1;
t = t * t % p;
}
return res;
}
int main() {
scanf("%d%d%d", &a, &b, &p);
printf("%d^%d mod %d=%d\n", a, b, p, fast_pow(a, b, p));
return 0;
}
时间复杂度 O ( log p ) O(\log p) O(logp),空间 O ( 1 ) O(1) O(1)。