[LintCode] Fast Power

快速幂运算的优化算法

最新推荐文章于 2019-01-11 00:34:18 发布

weixin_34087301

最新推荐文章于 2019-01-11 00:34:18 发布

阅读量55

点赞数

CC 4.0 BY-SA版权

原文链接：https://segmentfault.com/a/1190000004878046

本文介绍了快速幂运算的优化算法，通过分治法降低计算复杂度，特别适用于大规模整数的幂运算场景。

Problem

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

Example

For 2^31 % 3 = 2

For 100^1000 % 1000 = 0

Challenge

O(logN)

Note

应用求余公式: (a * b) % p = (a % p * b % p) % p
使用分治法，不断分解a^n为a^(n/2)，最终的子问题就是求解a^1或者a^0的余数。
唯一要注意的就是，若n为奇数，要将余数和a再代入求余公式，运算一次。

Solution

class Solution {
    public int fastPower(int a, int b, int n) {
        if (n == 0) return 1 % b;
        if (n == 1) return a % b;
        long product = fastPower(a, b, n/2);
        product = product * product % b;
        if (n % 2 == 1) product = product * a % b;
        return (int) product;
    }
}