6_8快速N次方

2017-10-23  本文已影响12人  X_Y

如果更快的求一个整数k的n次方。如果两个整数相乘并得到结果的时间复杂度为O(1),得到整数k的N次方的过程请实现时间复杂度为O(logN)的方法。

给定k和n,请返回k的n次方,为了防止溢出,请返回结果Mod 1000000007的值。

测试样例:
2,3
返回:8

class QuickPower {
public:
    int getPower(int k, int N) {
        // write code here 
        if(0==N) return 1;
        if(0==k) return 0;
        int mod = 1000000007;
        long res = 1;
        long tmp = k;
        while(N != 0){
            if((N & 1) == 1){
                res *= tmp;
            }
            N >>= 1;
            tmp = (tmp*tmp)%mod;
            res %= mod;
        }
        return (int)res;
    }
};
上一篇 下一篇

猜你喜欢

热点阅读