剑指 offer 刷题——16. 数值的整数次方

一、题目

给定一个 double 类型的浮点数 base 和 int 类型的整数 exponent。求 base 的 exponent 次方。

保证 base 和 exponent 不同时为 0。

二、思路

我们首先要考虑到的是，幂运算时底数不能为 0。

其次，题目里的 exponent 是 int 型的，整数要分成正整数、0 和 负整数三种情况。任何数的 0 次方结果为 1；负整数次方的结果可以用正整数次方的结果来计算，如 $2^{-3}=\frac{1}{2^3}$。所以我们只需计算 base 的正整数次方。

思路一：乘法（简单但不够高效）

参考代码：,

class Solution {
    public double myPow(double x, int n) {
        // 底数不能为零
        if(x == 0.0)
            return 0.0;
        // 任何数的 0 次方为 1
        if(n == 0)
            return 1.0;
        if(n == 1)
            return x;

        // 判断指数的正负
        int ex = n;
        if(ex < 0)
            ex = -ex;

        // 计算 x 的正整数次方
        double result = 1.0;
        for(int i = 1; i <= ex; i++){
            result *= x;
        }
        return n > 0 ? result : (1 / result);
    }
}

这种实现简单直观，但并不高效。如果输入的 exponent 为 32，就要在循环中做 31 次乘法。

思路二：位运算

如果我们的目标是求一个数的 32 次方，倘若我们已经知道了这个数的 16 次方，只需要再平方一次就可以了，而 16 次方的结果由 8 次方结果的平方得到。以此类推，我们求 32 次方只需要做 5 次乘法，即，先平方，再 4 次方，再 8 次方，再 16 次方，最后 32 次方。

当然，一个数的 n 次方也要分 n 为奇数和偶数两种情况：

• $a^n=a^{n/2}\times a^{n/2}$，n 为偶数；
• $a^n=a^{(n-1)/2}\times a^{(n-1)/2}\times a$，n 为奇数。

这个公式可以通过递归实现。

参考代码：

class Solution {
    public double myPow(double x, int n) {
        // 底数不能为零
        if(x == 0.0)
            return 0.0;
        
        int ex = n;
        if(ex < 0)
            ex = -ex;
        double result = powWithPositive(x, ex);
        
        return n >= 0 ? result : (1 / result);
    }

    // 一个数的正整数次方
    public double powWithPositive(double x, int n){
        if(n == 0)
            return 1.0;
        if(n == 1)
            return x;
        
        double result = powWithPositive(x, n >>> 1);
        result *= result;
        // n 为奇数
        if((n & 1) == 1)
            result *= x;
            
        return result;
    }
}

上面的实现方法相当于，在求 m 的 n 次方时，把 n 拆为几个整数的和来进行计算，只是这几个整数比较特殊（16、8、4、2、1）。相当于把 n 表示为二进制数，再按照二进制位拆分成几个数的和。比如，要算 $m^{13}$，把 13 拆成二进制数的和，即 $$m^{1101}=m^{1000+0100+0000+0001}=m^{1000}\times m^{0100}\times m^{0001}$$

所以，我们可以通过 & 1 和 >>>1 来逐位读取 1101，若当前位为 1，则将该位代表的乘数累乘到最终结果中。

参考代码：

public class Solution {
    public double Power(double base, int exponent) {
        if(base == 0)
            return 0.0;
        
        int n = exponent;
        if(n < 0)
            n = -n;
        double result = powWithPositive(base, n);
        
        return exponent > 0 ? result : (1 / result);
    }
    
    public double powWithPositive(double base, int exponent){
        if(exponent == 0)
            return 1.0;
        if(exponent == 1)
            return base;
        
        double factor = base;
        double res = 1.0;
        while(exponent != 0){
            if((exponent & 1) == 1)
                res *= factor;
            exponent = exponent >>> 1;
            factor *= factor;
        }
        
        return res;
    }
}