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

题目描述

给你三个整数 a,b,p，求 a^b mod p。

输入格式

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

输出格式

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

输入输出样例

输入 #1

2 10 9

输出 #1

2^10 mod 9=7

说明/提示

样例解释

2^10=1024，1024mod9=7。

数据规模与约定

对于 100% 的数据，保证 0≤a,b<2^31，a+b>0，2≤p<2^31。

#include<iostream>

using namespace std;

long long quickMod(long long b, long long p, long long k) {
	long long ans = 1;
	while (p > 0) {
		if (p % 2 == 1) {
			ans = ((ans % k) * (b % k)) % k;
		}
		b = (b % k) * (b % k);
		p /= 2;
	} 
	return ans;
}

int main() {
	long long b, p, k;
	cin >> b >> p >> k;
	long long res = quickMod(b, p,  k);
	cout << b << "^" << p << " mod " << k << "=" << res << endl; 
	return 0;
 }