LeetCode 372. Super Pow

最新推荐文章于 2025-05-19 04:15:00 发布

原创最新推荐文章于 2025-05-19 04:15:00 发布 · 340 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LeetCode 专栏收录该内容

42 篇文章

订阅专栏

本文介绍了一个计算 a 的 b 次方再对 1337 取模的方法。通过将 b 表示为数组的形式，并利用快速幂算法进行优化，解决了当 b 是一个极大数时的计算问题。提供了两个示例，分别为 a=2, b=[3] 结果为 8 和 a=2, b=[1,0] 结果为 1024。

Your task is to calculate ab mod 1337 where a is a positive integer and b is an extremely large positive integer given in the form of an array.

Example1:

a = 2
b = [3]

Result: 8

Example2:

a = 2
b = [1,0]

Result: 1024

class Solution {
public:
    const int base =1337;
    int superPow(int a, vector<int>& b) {
        int res = 1;
        int n = b.size();
        a = a % base;
        for (int i = 0; i < n; ++i) {
            res = (helper(res, 10) * helper(a, b[i])) % base;
        }
        return res;
    }
private:
    int helper(int a, int k) {         //a^k mod 1337 where 0 <= k <= 10
        int res = 1;
        for (int i = 0; i < k; ++i) {
            res = (res * a) % base;
        }
        return res;
    }
};