Fast Power 解题报告

Fast Power

Description

Calculate the an % b where a, b and n are all 32bit integers.

Example

For 231 % 3 = 2
For 1001000 % 1000 = 0

Challenge

O(logn)

实现思路

最直接的思路是直接进行n次循环，每次执行:
a = (a*a) % b运算，时间复杂度为o(n)。
除去模运算，我们发现：
1. 当n为奇数：
a^n = (a^(n-1)/2)^2 * a
2. 当n为偶数：
a^n = (a^n/2)^2
综合得:
a^n = (a^n/2)^2 * ((n&1) ? a:1) `
基于这个式子，我们可以递归计算，每次对平方结果进行平方，同时n/=2。
注意我们下面的代码是模拟这个求解过程，开始传入n,但我们算的时候是从((a^1)^2……)^2……,算起的，即从n/2=1开始算起，故我们先递归到最深层，再依次返回计算结果进行求解，具体看pow函数代码

class Solution {
    /*
     * @param a, b, n: 32bit integers
     * @return: An integer
     */
    public int fastPower(int a, int b, int n) {
        // write your code here
        if(a == 0 || b == 1){
            return 0;
        }

        return (int)pow(a,b,n);
    }

    public long pow(int a, int b, int n) {
        if(n == 0)
            return 1;
        if(n == 1)
            return a % b;
        long ret = pow(a,b,n/2);
        ret = (ret*ret) % b;
        if((n & 1) == 1)
            ret = (ret * a) % b;
        return ret;
    }
};