力扣 372. 超级次方

题目

你的任务是计算 ab 对 1337 取模，a 是一个正整数，b 是一个非常大的正整数且会以数组形式给出。

示例

输入：a = 2, b = [3]
输出：8

输入：a = 2, b = [1,0]
输出：1024

输入：a = 1, b = [4,3,3,8,5,2]
输出：1

输入：a = 2147483647, b = [2,0,0]
输出：1198

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/super-pow
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：递归

1、求幂的过程变成递归

2、求模的过程简化
对乘法的结果求模，等价于先对每个因子都求模，然后对因子相乘的结果再求模。

Python实现

class Solution:
    def superPow(self, a: int, b: List[int]) -> int:
        def mypow(a,k):
            res=1
            if k==0: return 1
            a%=1337
            for i in range(k):
                res*=a
                res%=1337
            return res

        if len(b)==0 or a==1: return 1
        var=b.pop()
        num1=mypow(a,var)
        num2=mypow(self.superPow(a,b),10)
        return (num1*num2)%1337

Java实现

class Solution {
    public int superPow(int a, int[] b) {
        int n = b.length;
        Stack<Integer> stack = new Stack<>();
        for (int num : b) stack.push(num);

        return hpPow(a, stack);
    }

    public int hpPow(int a, Stack<Integer> stack) {
        if (stack.isEmpty()) return 1;

        int var = stack.pop();
        int num1 = mypow(a, var);
        int num2 = mypow(hpPow(a, stack), 10);

        return (num1 * num2) % 1337;
    }

    public int mypow(int a, int k) {
        if (k == 0) return 1;
        int res = 1;
        a %= 1337;
        res %= 1337;

        for (int i = 0; i < k; i++) {
            res *= a;
            res %= 1337;
        }
        return res;
    }
}