1. 问题描述:
你的任务是计算 a ^ b 对 1337 取模,a 是一个正整数,b 是一个非常大的正整数且会以数组形式给出。
示例 1:
输入:a = 2, b = [3]
输出:8
示例 2:
输入:a = 2, b = [1,0]
输出:1024
示例 3:
输入:a = 1, b = [4,3,3,8,5,2]
输出:1
示例 4:
输入:a = 2147483647, b = [2,0,0]
输出:1198
提示:
1 <= a <= 2 ^ 31 - 1
1 <= b.length <= 2000
0 <= b[i] <= 9
b 不含前导 0
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/super-pow
2. 思路分析:
分析题目可以知道我们可以使用快速幂计算a ^ b,因为b给出的是数组的形式所以我们可以使用下面的公式计算即可,由公式可知这是一个递推的式子所以我们可以使用递归划分为若干个的子问题进行求解即可:
3. 代码如下:
from typing import List
class Solution:
# 计算快速幂
def quickPower(self, a: int, b: int, p: int):
res = 1
while b:
# 奇数的时候更新答案
if b & 1:
res = res * a % p
a = a * a % p
b >>= 1
return res
# 使用递归求解
def superPow(self, a: int, b: List[int]) -> int:
if not b: return 1
k = b.pop()
return self.quickPower(self.superPow(a, b), 10, 1337) * self.quickPower(a, k, 1337) % 1337