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为偶数,an =an/2 · an/2
当n为奇数,an = 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;
}
};