Leetcode 372. 超级次方 C++

Leetcode 372. 超级次方

题目

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

测试样例

示例 1:

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

示例 2:

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

题解

快速幂
我们先来看一看a^[1,5,6,4]，我们可以把幂指数进行缩减，变为 a⁴ * (a^[1,5,6])¹⁰ = a⁴ * ((a¹⁰)⁶*(a^[1,5])¹⁰⁰) = ``````。即通过这种方式循环相乘，我们需要用一个变量记录a¹⁰ⁱ。详细过程见代码

代码

int abPower(int a,int b,int mod){
        if(!b)  return 1;
        int ans = a;
        b--;
        while(b){
            ans = (ans*a) % mod;
            b--;
        }
        return ans;
    }
    int superPow(int a, vector<int>& b) {
        int n = b.size(),mod=1337;
        a = a % mod;
        int ans = 1,x=a;
        for(int i=n-1; i>=0; i--){
            ans = (ans*abPower(x,b[i],mod))%mod;
            x = abPower(x,10,mod);	//记录a^10(n-i)
        }
        return ans;
    }

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