剑指OFFER-数值的整数次方

Question

给定一个double类型的浮点数base和int类型的整数exponent。求base的exponent次方。
保证base和exponent不同时为0
关键词：快速幂 递归 二进制

Solution

暴力求幂

时间复杂度：O(N)
空间复杂度：O(1)

• Python

class Solution:
    def Power(self, base, exponent):
        # 只讨论简单的几种情况，其他如0^0、0^-1等情况未讨论
        if base==0:
            return 0
        if exponent==0:
            return 1
        ans = 1
        for i in range(abs(exponent)):
            ans *= base
        return ans if exponent>0 else 1/ans

• C++

class Solution {
public:
    double Power(double base, int exponent) {
        if(base==0) return 0;
        if(exponent==0) return 1;
        double ans = 1;
        int e = abs(exponent);
        for(int i=0; i<e; i++) 
            ans *= base;
        return exponent>0?ans:1/ans;
    }
};

递归

当n为偶数，aⁿ =a^n/2 · a^n/2
当n为奇数，aⁿ = a^(n-1)/2 · a^(n-1)/2 · a
时间复杂度：O(log(N))
空间复杂度：递归调用的空间

• Python

class Solution:
    def Power(self, base, exponent):
        if base==0:
            return 0
        if exponent==0:
            return 1
        e = abs(exponent)
        ans = self.Power(base, e>>1)
        ans *= ans
        if e&1:
            ans *= base
        return ans if exponent>0 else 1/ans

• C++

class Solution {
public:
    double Power(double base, int exponent) {
        if(base==0) return 0;
        if(exponent==0) return 1;

        double ans;
        int e = abs(exponent);
        ans = Power(base, e>>1);
        ans *= ans;
        if(e&1)
            ans *= base;
        return exponent>0? ans : 1/ans;
    }
};

快速幂

将指数拆成n/2的累加，如：
3 ^ 999
= 3 ^ (512 + 256 + 128 + 64 + 32 + 4 + 2 + 1)
相当于将999转成二进制：1111100111，每位代表要乘的倍数。
即，仅计算到3^512，途中若该位是1，则累乘当前值。

时间复杂度：O(N/2)
空间复杂度：O(1)

• Python

class Solution:
    def Power(self, base, exponent):
        if base==0:
            return 0
        if exponent==0:
            return 1
        
        ans = 1
        cur = base
        e = abs(exponent)
        while e:
        	if e & 1:
        		ans *= cur
        	cur *= base
        	e = e >> 1

        return ans if exponent>0 else 1/ans

• C++

class Solution {
public:
    double Power(double base, int exponent) {
        if(base==0) return 0;
        if(exponent==0) return 1;

        double ans = 1;
        double cur = base;
        int e = abs(exponent);
        while(e){
            if(e&1)
                ans *= cur;
            cur *= base;
            e = e >> 1;
        }
        return exponent>0? ans : 1/ans;
    }
};