递归_medium_50_Pow(x, n)

题目描述

思路：

• 类似转成二进制的方法，比如求 3 的 15 次方，即$3^{15}=3^{1+2+4+8}=3^1\ast 3^2\ast 3^4\ast 3^8$，10转成二进制为1111
• 这样我们在计算时就可以利用上次的结果，计算过程类似二分。
• 时间复杂度 $O(logn)$，空间复杂度 $O(1)$

转成二进制解法

class Solution:
    def myPow(self, x: float, n: int) -> float:
        if not x:
            if n >= 0: return 0  # x 为0时 直接输出0
            else: return float("inf")
        """
        思路：
            先转成二进制
            然后类似于(二分)层层递进，求当前位置的结果
        """
        key = False  # 是否为负数
        if n < 0:
            key = True
            n = -n
        # 转二进制
        res = []  # 存储的是 二进制的逆序
        while n:
            if not n%2:  # 余数为0
                res.append('0')
            else:
                res.append('1')
            n //= 2
        result = 1
        for one in res:
            if one == '1':
                result *= x
            x *= x
        return result if not key else 1/result

• 当然，我们无需求出二进制的结果，可以直接边求边计算
• 下面分别给出 递归 和 迭代的代码

递归解法

class Solution:
    def myPow(self, x: float, n: int) -> float:
        def quickMul(N):
            if N == 0:
                return 1.0
            y = quickMul(N // 2)
            return y * y if N % 2 == 0 else y * y * x
        
        return quickMul(n) if n >= 0 else 1.0 / quickMul(-n)

• 时间复杂度 $O(logn)$，空间复杂度 $O(logn)$，因为需要栈空间

迭代解法

class Solution:
    def myPow(self, x: float, n: int) -> float:
        if n < 0:
            x = 1/x
            n = -n
        mid_res = x
        res = 1
        while n:
            if n%2:  # == 1
                res *= mid_res
            n //= 2
            mid_res = mid_res*mid_res
        return res

• 时间复杂度 $O(logn)$，空间复杂度 $O(1)$