372.超级次方

372.超级次方_第1张图片

其实这个题总的来说并不是特别难,虽然题解有一些数学公式看起来有点唬人。思路很好想,就顺序或者倒序遍历数组b,遍历的时候注意b对应的位数会随着变,例如使用倒序,那么b的位数每次遍历会相应的增加,要把之前的结果和这次的结果相乘得到这次的总结果,但是会有一个问题比较麻烦,就是每次遍历中a^n,这个n如果我们用b[i]乘以对应的位数的话会很大,尤其在数组b很长的时候会超出int范围,这个时候题解关键的地方就来了
372.超级次方_第2张图片
就是我们不要把位数放到b[i]去算,而是放到底数a来算,那有什么区别呢,放到a来算不也是很大的吗,甚至位数本身就已经超出int范围了,但是放到a来算,我们要知道取余是满足分配律的,所以我们可以用改进后的快速幂算法来计算这个结果,这样就不会溢出了,关于快速幂可看LeetCode第50题。所以代码如下:

class Solution {
private:
    int quickPow(int x, int n) {
        if (n == 0) {
            return 1;
        }
        int pre = quickPow(x, n / 2) % 1337;
        return n % 2 == 0 ? pre * pre : pre * pre % 1337 * x;
    }
public:
    int superPow(int a, vector<int>& b) {
        if (a >= 1337) {
            a = a % 1337;
        }

        int ans = 1;
        for (int i = b.size() - 1; i >= 0; --i) {
            ans = ans * (quickPow(a, b[i]) % 1337) % 1337;
            a = quickPow(a, 10) % 1337;
        }
        return ans;
    }
};

这个题的关键就在于取余满足分配律,对底数取余不影响结果,所以只要是底数的地方都可以取余。

你可能感兴趣的:(LeetCode刷题笔记,数据结构,算法,c++,LeetCode)