Super Pow

前提知识： ab % k = (a%k)(b%k)%k

对b做二分处理，即 (a^b) % k = (a * a)^(b/2) % k，这样时间复杂度就是logb了

当然还要处理b是奇数的情况，而且b在这道题里是个数组，所以要写一些针对数组的运算函数

class Solution {
public:
	int superPow(int a, vector<int>& b) {
		int ans = 1;
		a = a % 1337;
		while (morethanzero(b))
		{
			if (IsOdd(b))
				ans = (ans * a) % 1337;
			div(b, 2);
			a = (a * a) % 1337;
		}
		return ans;
	}

	void div(vector<int> &x, int y)
	{
		int tmp = 0;
		for (int i = 0; i < x.size(); i++)
		{
			x[i] += tmp * 10;
			tmp = x[i] % y;
			x[i] = x[i] / y;
		}
	}

	bool IsOdd(const vector<int> & nums)
	{
		if (nums.back() & 1)
			return true;
		else
			return false;
	}

	bool morethanzero(vector<int> & x) {
		for (int i = x.size() - 1; i >= 0; i--) {
			if (x[i] > 0) return true;
		}
		return false;
	}
};

然而还是超时了，囧，因为每次做除法的时候都要遍历一遍数组，所以时间复杂度实际上是nlogb

最后用了官方discuss里的一份代码，简单解释一下：

a^1234567 % k = (a^1234560 % k) * (a^7 % k) % k = (a^123456 % k)^10 % k * (a^7 % k) % k

写成函数的形式就是：f(a,1234567) = f(a, 1234560) * f(a, 7) % k = f(f(a, 123456),10) * f(a,7)%k

所以用迭代的方法进行计算

class Solution {
    const int base = 1337;
    int powmod(int a, int k) //a^k mod 1337 where 0 <= k <= 10
    {
        a %= base;
        int result = 1;
        for (int i = 0; i < k; ++i)
            result = (result * a) % base;
        return result;
    }
public:
    int superPow(int a, vector<int>& b) {
        if (b.empty()) return 1;
        int last_digit = b.back();
        b.pop_back();
        return powmod(superPow(a, b), 10) * powmod(a, last_digit) % base;
    }
};