LeetCode 第 372 题 (Super Pow)

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

这道题与第 50 题其实大同小异。唯一一个值得注意的地方是 $1336^3$ 超过了 int 型变量可以表示的整数的范围。所以我们的程序中不能出现 (a * b * c) % 1337 这样三个数字相乘的代码。

下面是我的代码：

int superPow(int a, vector<int>& b)
{
    int z = 1, a2, a4;
    int size = b.size() - 1;
    a = a % 1337;
    for(int i = size; i >=0; i--)
    {
        a2 = (a * a) % 1337;
        a4 = (a2 * a2) % 1337;
        switch(b[i])
        {
        case 1:
            z = (z * a) % 1337;
            break;
        case 2:
            z = (z * a2) % 1337;
            break;
        case 3:
            z = (z * a2) % 1337;
            z = (z * a) % 1337;
            break;
        case 4:
            z = (z * a4) % 1337;
            break;
        case 5:
            z = (z * a4) % 1337;
            z = (z * a) % 1337;
            break;
        case 6:
            z = (z * a4) % 1337;
            z = (z * a2) % 1337;
            break;
        case 7:
            z = (z * a4) % 1337;
            z = (z * a2) % 1337;
            z = (z * a) % 1337;
            break;
        case 8:
            z = (z * a4) % 1337;
            z = (z * a4) % 1337;
            break;
        case 9:
            z = (z * a4) % 1337;
            z = (z * a4) % 1337;
            z = (z * a) % 1337;
            break;
        default:
            break;
        }
        a = (a4 * a4) % 1337;
        a = (a * a2) % 1337;
    }
    return z;
}