快速幂

最新推荐文章于 2024-05-21 23:34:41 发布

gerrylon007

最新推荐文章于 2024-05-21 23:34:41 发布

阅读量141

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/butterfly5211314/article/details/86294458

版权

算法专栏收录该内容

24 篇文章 0 订阅

订阅专栏

问题

求 $a^b$ , 为了简单起见, 假定 $b\ge0 且b为整数$ .

朴素求解方法

这个问题很简单, 最简单的想法, a自乘b次即可, 也就是所谓的朴素求解方法:

int pow_naive(int a, int b) {
    int ret = 1;
    
    while (b--) {
        ret *= a;
    }
    return ret;
}

上述方法时间复杂度为 $O (n)$ .

快速幂求解

而"快速幂"方法可以将时间复杂度提升到 $O(\log n)$ , 它基于这样的想法:
求 $a^b$ , 将b展开为二进制表达, 如当b=13时, 有:

$(1101)_2 = 1 \times 2^0 + 0 \times 2^1 + 1 \times 2^2 + 1 \times 2^3$
$a^b = a^{2^0 + 0 + 2^2 + 2^3} = a^{2^0} \times a^{2^2} \times a^{2^3}$

代码如下:

int pow_fast(int a, int b) {
    int ret = 1;
    int base = a;

    while (b) {
        if (b & 1) { // 当前b最低位为1, 比如 1101最右边的1
            ret *= base;
        }
        b >>= 1;
        base *= base;
    }
    return ret;
}

应用示例

可以看一个相关的应用来体会一下用法:
杭电快速幂模基础板题
题目简单描述为: $1\le N \le 1,000,000,000, 求N^N的最后一位数$ .

简单分析下:
显然, 直接算太大, 会整型溢出. 假定结果为M, 最后一位数不就是 M%10嘛.
其实这就是典型的快速幂取模问题.

要使用快速幂取模来解决上述问题, 需要以下公式:
$a^b \mod c = (a \mod c)^b \mod c$

证明如下:

首先证明:
$\mod c = ((a \mod c) \times (b \mod c)) \mod c$

证明过程：
$\mod c = r_{1} \Rarr a = q_{1}c + r_{1} \\ b \mod c = r_{2} \Rarr b = q_{2}c + r_{2} \\ \begin{aligned} ab \mod c &= (q_{1}c + r_{1})(q_{2}c + r_{2}) \mod c \\ &= [q_{1}q_{2}c^2 + r_{1}r_{2} + (q_{1}r_{2} + q_{2}r_{1})c] \mod c \\ &= r_{1}r_{2} \mod c \\ &= ((a \mod c) \times (b \mod c)) \mod c \end{aligned}$
证毕.
也就是说: 积的余 = 余的积的余
再证明:
$a^b \mod c = (a \mod c)^b \mod c$

证明过程：
$\begin{aligned} a^b \mod c &= (a \times a^{b-1}) \mod c \\ &= [(a \mod c) \times (a^{b-1} \mod c)] \mod c \\ &= [(a \mod c) \times (a \mod c) \times (a^{b - 2} \mod c)] \mod c \\ &= ... \\ &= (a \mod c)^{b} \mod c \end{aligned}$

有了这个理论, 代码就不难了:

// 快速幂取模 模板代码
int pow_fast_tpl(int a, int b, int c) {
    int ret = 1;
    int base = a % c;

    while (b) {
        if (b & 1) {
            ret = (ret * base) % c;
        }
        base = (base * base) % c;
        b >>= 1;
    }

    return ret;
}