LeetCode 372. 超级次方

最新推荐文章于 2024-10-01 05:04:32 发布

XY_xyz_xyz

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量207

点赞数

文章标签： leetcode 蓝桥杯算法

本文链接：https://blog.csdn.net/XY_xyz_xyz/article/details/121843492

版权

快速幂大数运算模运算数组表示算法

关键词由CSDN通过智能技术生成

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/super-pow

解法：

快速幂。这题是快速幂，求模，指数分解的灵活运用。

首先乘法求模满足分配律：

$(a*b)%c=(a%c)*(b%c)$ ；

其次由数组表示的整数n可表示为 $n=\sum_{i=0}^{m-1}(10^{m-1-i}*b_{i})$ .

最终指数乘法 $a^{x+y}=a^{x}*a^{y}$ , $a^{xy}=(a^{x})^{y}$ . 则对于每个i， $a^{(10^{m-1-i})*b_{i}}=(a^{10^{m-1-i}})^{b_{i}}$

如n=10，则 $a^{n}=a^{(10^{2-1-0}*1+10^{2-1-1}*0)}=a^{(10^{2-1-0}*1)}*a^{(10^{2-1-1}*0)}$

$=(a^{(10^{2-1-0})})^{1}*(a^{(10^{2-1-1})})^{0}$

可以发现 $(a^{10^{m-1-i}})$ 这一部分可以倒序，每次求个10次方可得。、

class Solution {
public:
	int superPow(int a, vector<int>& b) {
		const int MOD = 1337;
		int len = b.size();
		auto SP = [&](long long x, int y)
		{
			long long sum = 1;
			while (y)
			{
				if (y & 1)
					sum = ((sum%MOD)*(x%MOD)) % MOD;
				x = ((x%MOD)*(x%MOD)) % MOD;
				y = y >> 1;
			}
			return sum;
		};
		long long ans = 1;
		long long temp = a;
		for (int i = len - 1; i >= 0; --i)
		{
			ans = ((ans%MOD)*(SP(temp, b[i]) % MOD)) % MOD;
			temp = SP(temp, 10);
		}
		return ans;
	}
};