快速幂 c++

小王子009

已于 2024-01-26 22:05:06 修改

阅读量852

点赞数 3

文章标签： c++ 算法数据结构快速幂取模

于 2023-09-13 13:36:06 首次发布

本文链接：https://blog.csdn.net/weixin_71529971/article/details/132850866

版权

一般大家写 $x^a$ 都是这样：

int ans = 1;
for (int i = 1; i <= a; i ++)
    ans *= x;

时间复杂度 $O(n)$

但是，这对于我们还不够！我们要 $O(logn)$

首先我们得知道一个数学知识：

$x^{a^{b}} = x^{a*b}$

那么求 $x^a$ ，就有以下递归式：

如果 a 能被2整除（为偶）

$x^a = x^{(a/2)^{2}} = x^{a/2} * x^{a/2}$

如果 a 不能被2整除（为奇）

$x^a = x^{(a/2)^{2}} * x = x^{a/2} * x^{a/2} * x$ (这里a/2是整除)

每次求 $x^a$ 都先求调用 $x^{a/2}$ ，不就是 $O(logn)$ 么？

最后补充一个东西：

$x^a mod b = (x^{i} mod b * x^{j} mod b) mod b$ (i + j = a)

我们这里一边乘一边模，可以防止运行时超出 longlong

代码：

#include <iostream>
using namespace std;
typedef long long LL;
LL a, b, m;
//m是取模的数
LL q_pow(LL a, LL b, LL m) {
	if(b == 0)
		return 1;
	LL tmp = q_pow(a, b >> 1, m) % m;
	return (b & 1 ? a : 1) * tmp % m * tmp % m;
//b & 1 和 b % 2 == 1 是等价的
}
int main() {
	cin >> a >> b >> m;
	cout << q_pow(a, b, m);
	return 0;
}