1、题目描述
给定一个double类型的浮点数base和int类型的整数exponent。求base的exponent次方。
2、代码详解
2.1 精简版写法
快速幂解析(分治法角度)
https://leetcode-cn.com/problems/powx-n/
详细注释见下面的版本
class Solution:
def myPow(self, x, n):
# 特例底数为0且指数为负(0不能求倒数)
if x == 0 and n < 0:
return ZeroDivisionError
if x == 0.0:
return 0.0
res = 1
if n < 0:
x, n = 1 / x, -n
while n:
# 用“&与”来代替“%2”,判断奇偶数
if n & 1:
res *= x # 如果求奇数次幂,多乘上一次x
x *= x
n >>= 1 # 用右移操作,代替“除以2”, 如2的5次方,可以先求2的2次方(5//2=2)
return res
时间复杂度 O(logn) : 二分的时间复杂度为对数级别。
空间复杂度 O(1): 变量占用常数大小额外空间。
补充:快速幂
快速幂解析(二进制角度)解析 https://leetcode-cn.com/problems/powx-n/solution/50-powx-n-kuai-su-mi-qing-xi-tu-jie-by-jyd/
2.2 牛客剑指版
先说结论:
- 特例:底数为0且指数为负,抛异常“0不能求倒数”
- equal_zero时,因为计算机内表示小数有误差,要写成abs(num - 0.0) < 0.0000001
- 1.0/base,不能写成1/base
- 位运算优化效率
- “整除以2(//2)” 写成 “右移1位(>>1)”
- “判断为奇数(%2)” 写成 “与操作(& 1 == 1)”
# -*- coding:utf-8 -*-
class Solution:
def Power(self, base, exponent):
# 特例底数为0且指数为负(0不能求倒数)
if self.equal_zero(base) and exponent < 0:
return ZeroDivisionError
result = self.power_value(base, abs(exponent))
if exponent < 0:
return 1.0/result
else:
return result
# 判断底数base是不是等于0的时候,不能直接写base==0
# 因为计算机内表示小数时有误差,只能判断他们的差的绝对值是不是在一个很小的范围内
def equal_zero(self, base):
if abs(base - 0.0) < 0.0000001:
return True
def power_value(self, base, exponent):
if exponent == 0:
return 1
if exponent == 1:
return base
result = self.power_value(base, exponent >> 1) # 用右移操作,代替“除以2”, 如2的5次方,可以先求2的2次方(5//2=2)
result *= result
# 如果求奇数次幂
# 用“&与”来代替“%2”,判断奇偶数
if exponent & 1 == 1:
result = result * base
return result
S = Solution()
print(S.Power(2, 5))
print(S.Power(5, -2))
print(S.Power(0, -2)) # 异常特例
2.3 力扣版
实现 pow(x, n) ,即计算 x 的 n 次幂函数。
class Solution(object):
def myPow(self, base, exponent):
# 特例底数为0且指数为负(0不能求倒数)
# if self.equal_zero(base) and exponent < 0:
if base == 0 and exponent < 0:
return ZeroDivisionError
result = self.power_value(base, abs(exponent))
if exponent < 0:
return 1.0 / result
else:
return result
# 判断底数base是不是等于0的时候,不能直接写base==0
# 因为计算机内表示小数时有误差,只能判断他们的差的绝对值是不是在一个很小的范围内
# def equal_zero(self, base):
# if abs(base - 0.0) < 0.0000001:
# return True
def power_value(self, base, exponent):
if exponent == 0:
return 1
if exponent == 1:
return base
result = self.power_value(base, exponent >> 1) # 用右移操作,代替“除以2”, 如2的5次方,可以先求2的2次方(5//2=2)
result *= result
# 如果求奇数次幂
# 用“&与”来代替“%2”,判断奇偶数
if exponent & 1 == 1:
result = result * base
return result
s = Solution()
print(s.myPow(2, 5))