372. 超级次方

你的任务是计算 ab 对 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

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

class Solution {

    private final int MOD = 1337;

    public int superPow(int a, int[] b) {
        int ans = 1;
        int base = a % MOD;
        for (int i = b.length - 1; i >= 0; --i) {
            ans = (ans * quickPow(base, b[i])) % MOD;
            base = quickPow(base, 10);
        }
        return ans;
    }

    private int quickPow(int x, int n) {
        int ans = 1;
        int base = x;
        while (n > 0) {
            if ((n & 1) == 1) {
                ans = ans * base % MOD;
            }
            n >>= 1;
            base = base * base % MOD;
        }
        return ans;
    }
}
posted @ 2022-01-05 21:08  Tianyiya  阅读(44)  评论(0)    收藏  举报