快速幂详解

快速幂

力扣50. Pow(x, n)

力扣LCR 134. Pow(x, n)

题意：

求$x^n$

• -100.0 < x < 100.0
• -$2^{31}$ <= n <= $2^{31-1}$
• n 是一个整数
• 要么 x 不为零，要么 n > 0
• $-10^4$ <= $x^n$ <= $10^4$

---

思路一：循环将x乘n次，龟速幂。但是当n很大时，会因迭代次数过多而超时。

思路二：二进制优化。

对于正整数n，可以分解为$n=2^{n_1}+2^{n_2}+...+2^{n_k}$。即$x^n=x^{2^{n_1}+2^{n_2}+...+2^{n_k}}=x^{2^{n_1}}{x}^{2^{n_2}}...x^{2^{n_k}}$。

例如n=5，二进制表示为0101，观察1的位置（分别是第0、2位），所以$5=2^0+2^2$；若x=3，则$3^5=3^{2^0+2^2}=3^1{3}^4$。

对于n<0的情况，转化一下即可：$x^n\to (\frac{1}{x}{)}^{-n}$。注意当n=-2^31是，其负数是2^31超出了int范围的上界2^31-1，所以代码上要用long类型接收n。

这样的话，只要计算$3^1{3}^4$，而不用计算$3\ast 3\ast 3\ast 3\ast 3$。只需要最多32位即可表示int范围内的整数n，所以只需log(n)次运算就可以完成幂的计算。

关键在于如何完成转化：$3^5=3^{2^0+2^2}=3^1{3}^4$。我们可以发现拆分的项数是和二进制1的数量相同的，只需要在每次遇到二进制1时乘对应值即可，而对应的值即是累乘x。

如$3^5=3^1{3}^4$，$5=(0101{)}_b，ans=1$：

1. 对于第0位（1），$ans$乘上此时的对应值$3^{2^0}=3^1$，然后x累乘得第1位的对应值$3\ast 3=3^{2^1}=3^2$
2. 对于第1位（0），不需要计算答案，继续累乘x得第2位的对应值$3^2{3}^2=3^{2^2}=3^4$
3. 对于第2位（1），$ans$乘上此时的对应值$3^4$；
4. 对于第3位（0），不需要计算答案，继续累乘x得第2位的对应值
5. b的二进制位已遍历结束，最后$ans=3^1{3}^4$

---

显然遍历b的二进制位的时间复杂度为$log(n)$，代码如下：

// x^n
class Solution {
public:
    double myPow(double x, int n) {
        double ans = 1;   // 最终答案
        long b = n;		// long接受防止越界
        
        if(b < 0)	// 处理负数幂的情况
            x = 1 / x,b = -b;
        while(b){   // 遍历b的二进制位(移位)
            if(b & 1)  // 二进制位为1
                ans *= x;   // 累乘答案
            x *= x;   // 对应值
            b >>= 1;  // 右移(除2)
        }
        return ans;
    }
};