算法：模幂运算【快速幂】与力扣372.超级次方

算法：模幂运算【快速幂】与力扣372.超级次方

1、 题目

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

提示：

1 <= a <= 231 - 1
1 <= b.length <= 2000
0 <= b[i] <= 9
b 不含前导 0
链接：https://leetcode-cn.com/problems/super-pow

2、 解析

我们可以看出b的大小可以达到10^2000这么大，就算是long long也储存不了这么大的数，对于这么大的数，不可能使用常规的思路进行求解。

模幂运算

定理 (a * b ) % c == ((a % c) * (b % c)) % c;

在处理掉数的储存问题之后，还有一个问题就是，以常规的循环计算幂a^b,时间复杂度为O(b)， b是有可能比long

long大的，所以也不能以常规的方式计算。

快速幂

计算a^b时：

res = a ^ (b / 2) * a ^ (b / 2) , b / 2 == 0;

res = a* a ^ (b - 1) / 2 * a ^ (b - 1) / 2 , b / 2 == 1;

int mypow(int a, int b,int mod){
        int res = 1;
        a%=mod;
        while(b){
            if(b&1){
                res *= a;
                res %= mod;
            }
            a = (a * a)%mod;
            b>>=1;
        }
        return res;
    }

2 、题目解答

class Solution {
public:
    int mypow(int a, int b,int mod){
        int res = 1;
        a%=mod;
        while(b){
            if(b&1){
                res *= a;
                res %= mod;
            }
            a = (a * a)%mod;
            b>>=1;
        }
        return res;
    }
    int superPow(int a, vector<int>& b) {
        int res = 1;
        for(int i = b.size() - 1; i >=0; i--){
            res = res*mypow(a, b[i], 1337)%1337;
            a = mypow(a, 10, 1337)%1337;
        }
        return res;
    }

};