【洛谷】P1226 【模板】快速幂||取余运算

题目地址：

https://www.luogu.com.cn/problem/P1226

题目描述：
给你三个整数$a,b,p$，求$a^b\mathrm{m}\mathrm{o}\mathrm{d}p$。

输入格式：
输入只有一行三个整数，分别代表$a,b,p$。

输出格式：
输出一行一个字符串a^b mod p=s，其中$a,b,p$分别为题目给定的值，$s$为运算结果。

数据范围：
对于$100\%$的数据，保证$0\le a,b<2^{31}$，$a+b>0$，$2\le p<2^{31}$。

思路是快速幂（或者也可以称为叫“倍增法”），由$a$出发，依次计算$a^{2^n},n=0,1,2,...$，然后将$b$看成二进制，倍增地计算$a^b\mathrm{m}\mathrm{o}\mathrm{d}p$。代码如下：

#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(1)$。