【LeetCode】372. Super Pow 超级次方

作者： 负雪明烛
id： fuxuemingzhu
个人博客： http://fuxuemingzhu.cn/
公众号：负雪明烛

---

题目地址: https://leetcode.com/problems/super-pow/description/

题目描述：

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

Example 1:

Input: a = 2, b = [3]
Output: 8

Example 2:

Input: a = 2, b = [1,0]
Output: 1024

题目大意

实现a的b次方的函数。但是给出的b是超级巨大的，而且是用数组保存着每一位的。

解题方法

这个题是 50. Pow(x, n) 的拓展题，都是快速求幂的问题。但是这个题由于结果的数值大，需要模 1337；对于模什么数一般都是瞎选的，不用考虑这个题为什么模 1337。

这个题的难点在于：如何求数组表示的超级大的数字 b 次幂。
​

求幂也可以边遍历，边求。
​

比如求 $2^{2}3$ ，其中 23 用数组表示即 $[2,3]$。
​

那么：
​

$2^{23}=(2^{20})\ast (2^3)=(2^2{)}^{10}\ast (2^3)$ 。
​

也就是说把可以遍历数组：

• 先求 $2^2$，把它的结果求一次 10 次幂；
• 然后把计算结果，再乘以 $2^3$。

需要注意这里每次求幂计算之前和之后都要 % 1337 。是为了防止数值过大，导致求幂的结果溢出的问题。
​

先求幂再取模，与先取模再求幂得到的结果是一样的。
​

C++ 代码：

class Solution {
public:
    const int MOD = 1337;
    int superPow(int a, vector<int>& b) {
        int res = 1;
        for (int x : b) {
            res = pow(res, 10) * pow(a, x) % MOD;
        }
        return res;
    }
    
    int pow(int a, int b) {
        a %= MOD;
        if (b == 0) {
            return 1;
        }
        if (b % 2 == 1) {
            return a * pow(a, b - 1) % MOD;
        }
        return pow(a * a, b / 2) % MOD;
    }
};

Java 代码：

class Solution {
    final int MOD = 1337;
    public int superPow(int a, int[] b) {
        int res = 1;
        for (int x : b) {
            res = pow(res, 10) * pow(a, x) % MOD;
        }
        return res;
    }
    
    int pow(int a, int b) {
        a %= MOD;
        if (b == 0) {
            return 1;
        }
        if (b % 2 == 1) {
            return a * pow(a, b - 1) % MOD;
        }
        return pow(a * a, b / 2) % MOD;
    }
}

Python 代码：

class Solution(object):
    def superPow(self, a, b):
        """
        :type a: int
        :type b: List[int]
        :rtype: int
        """
        res = 1
        for x in b:
            res = self.pow(res, 10) * self.pow(a, x) % 1337
        return res
        
    def pow(self, a, b):
        if b == 0 or a == 1: return 1
        if b % 2:
            return a * self.pow(a, b - 1) % 1337
        return self.pow((a * a) % 1337, b / 2) % 1337

• 时间复杂度：$O(N)$，N 是数组的长度。
• 空间复杂度：$O(N)$，递归用到了栈的深度。

总结

1. 今天这个题目考了快速幂，偶尔会使用到的技巧，需要掌握。
2. 做这种结果很大，需要取模的题目时，一定要注意不要溢出。

参考资料：

http://www.cnblogs.com/grandyang/p/5651982.html

日期

2018 年 10 月 7 日 —— 假期最后一天！！
2021 年 12 月 5 日 —— 即将搬家，舍不得现在的房子啊