首页 > 编程知识 正文

快速幂取模,快速幂取模算法

时间:2023-05-05 19:13:47 阅读:264895 作者:2422

目录

1 问题描述

2 原因分析

3 解决方法

4 快速幂讲解

5 快速乘讲解

6 完整代码

7 References


1 问题描述

我们发现,在 int 型下使用 pow 函数求 ,结果为 124 。

2 原因分析

pow 函数的返回值为 double型 ,因浮点数长度问题,存在截断误差。

3 解决方法

将变量定义为 double 型

有没有更快求幂的方法?

4 快速幂讲解

 假设我们要求 ,按照朴素算法就是把 a 连乘 b 次,这样一来时间复杂度是 O(b) ,即是 O(n) 级别。但快速幂能做到 O(logn) 的复杂度。

快速幂:

对于二进制的位运算,我们需要用到 "&" 与 ">>" 运算符,详见 位运算符的应用

先上实现快速幂运算的具体代码:

long long ksm(long long a, long long b) { long long ans = 1, base = a; while(b != 0) {if(b & 1 != 0) { ans *= base;}base *= base;b >>= 1; } return ans;}

其中 “b & 1” 指取 b 的二进制数的最末位,如 11 的二进制数为 1011 ,第一次循环,取的是最右边的 “1” ,以此类推。

而 “b >>= 1” 等效于 b = b >> 1,即右移 1 位,删去最低位。

以 a ^ 11 为例

b 的二进制数为 1011,二进制从右向左算,但乘出来的顺序是 ,是从左向右的。我们不断的让 目的是累乘,以便随时对 ans 做出贡献。

要理解 这一步:因为 == ,下一步再乘,就是 () * () == ,然后同理 () * () == ,由此可以做到 → → → → → .......指数正好是  。再看上面的例子, = ,这三项就可以完美解决了,快速幂就是这样。

如还有不明白的地方,建议手动模拟代码的运行过程。

5 快速乘讲解

我们知道,在计算机中做加法运算会比乘法快得多(参考模电中的加法器),做乘法运算往往会溢出,即使用 long long 类型也拯救不了。因此需要寻找一种能高效完成乘法运算且不会溢出的算法,这就是快速乘算法。

快速乘与快速幂原理相似,也是将运算转换为二进制处理:

以 a * 11 为例:  11 的二进制数为 1011,则有

就是把快速幂中的 * 号改为+号

long long ksc(long long a, long long b) { long long ans = 0; while(b != 0) {if(b & 1 != 0) { ans += a;}a += a;b >>= 1; } return ans;}

此版本的复杂度和快速幂一样,也是 O(logn) 。如果需要特别卡常数,可以去了解 O(1) 版本的快速乘。

6 完整代码

为了防止溢出,一般快速幂和快速乘的算法会在 mod 下运用,下面给出取模运算代码。

#include <bits/stdc++.h>using namespace std;typedef long long ll;const ll mod = 1e7;//a ^ bll ksm(ll a, ll b, ll mod) { ll ans = 1, base = a; while(b != 0) {if(b & 1 != 0) { ans = (ans * base) % mod;}base = (base * base) % mod;b >>= 1; } return ans;}//a * bll ksc(ll a, ll b, ll mod) { ll ans = 0; while(b != 0) {if(b & 1 != 0) { ans = (ans + a) % mod;}a = (a + a) % mod;b >>= 1; } return ans;}int main() { cout << "5 ^ 3 = " << ksm(5, 3, mod) << endl; cout << "345352 * 11 = " << ksc(345352, 11, mod) << endl; return 0;}

 运算结果:

7 References https://www.cnblogs.com/CXCXCXC/p/4641812.html

以上,有问题欢迎指正!

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。