快速幂模板

Posted 2020-09-02 bennettz

以下代码输入a, b, p。会输出a的b次方再取余p。

#include <cstdio>
using namespace std;
int power(int a, int b, int p) {
    if (!b) return 1;
    long long t = power(a, b/2, p);
    if (b % 2 == 0) t = t * t % p;
    else t = t * t * a % p;
    return t;
}
int main () {
    int a, b, p;
    scanf("%d %d %d", &a, &b, &p);
    printf("%d\n", power(a, b, p));
    return 0;
}