快速幂算法-Pow(x,n)

棱角码农

已于 2022-12-25 11:06:35 修改

阅读量1.1k

点赞数 2

文章标签：算法 c++ leetcode

于 2021-12-06 15:19:41 首次发布

本文链接：https://blog.csdn.net/saxon_li/article/details/121746750

版权

一、递归

原理：

举个例子求解 $x^{77}$ ，不需要对 x 进行 77 次乘法计算，而可以按照 $x\rightarrow x^{2}\rightarrow x^{4}\rightarrow x^{9}\rightarrow x^{19}\rightarrow x^{38}\rightarrow x^{77}$ 的顺序，在 $x\rightarrow x^{2},x^{2}\rightarrow x^{4},x^{19}\rightarrow x^{38}$ 这些步骤中，可以直接把上次的结果进行平方，而在 $x^{4}\rightarrow x^{9},x^{9}\rightarrow x^{19},x^{19}\rightarrow x^{38}$ 这些步骤中，在对上次结果平方后，还需额外乘一个 x 。

从右往左推导上述过程，利用分治思想，可以得到快速幂算法递归步骤：

（1）求解 $x^{n}$ ，可以先递归的计算 $y= x^{n/2}$ ；

（2）根据递归的计算结果，若 n 为偶数，则 $x^{n}=y^{2}$ ；若 n 为奇数，则 $x=x*y^{2}$ ；

（3）递归出口为 n = 0，任意数的 0 次方均为 1 。

注意：当 n 为负数时，可以计算 $x^{-n}$ 再取倒数得到结果。

double quickMul(double x,long long N) {
    if(N==0) return 1.0;
    double y=quickMul(x,N/2);
    return N%2==0?y*y : y*y*x;//讨论 N 的奇偶
}

double myPow(double x, int n) {
    long long N=n;//INT_MIN > INT_MAX 故要用long long
    return N>0? quickMul(x,N):1.0/quickMul(x,-N);//讨论 N 的正负
}

二、迭代

原理：

由于递归需要使用额外的栈空间，故试着将递归转写为迭代。遇到一个问题，就是不知道什么时候需要额外乘一个 x 。需借助 整数的二进制拆分 :

$x^{n}=x^{2^{k}}*x^{2^{k-1}}*...*x^{2^{1}}*x^{2^{0}}$

从 x 开始不断进行平方，如果如果 n 的第 k 个（从右往左，从 0 开始计数）二进制位为 1，那么我们就将对应的贡献 $x^{2^{k}}$ 计入答案。具体快速幂迭代算法参考下方代码。

double myPow(double x, int n) {
    long long N=n;
    if(N<0) N=-N;
    double res=1.0;
    double x_contribution=x;//贡献值
    while(N>0)
    {
        if(N%2==1) res*=x_contribution;//二进制最低位为 1 ，需把贡献记入结果
        x_contribution*=x_contribution;
        N=N/2;//不断右移， 舍弃 N 二进制的最低位
    }
    if(n<0) return 1.0/res;            
    else return res;
}