算法小白,最近刷LeetCode。希望能够结合自己的思考和别人优秀的代码,对题目和解法进行更加清晰详细的解释,供大家参考^_^
50. Pow(x, n)
Implement pow(x, n).
题目让求x的n次幂,n可以是负数。这里只考虑n是整数的情况,负数幂的话就是1.0/x^n(n>0)嘛。
如果按照正常的逻辑,将x乘上n次,显然是会超时的。题目标签说是一个二分搜索问题,那就从二进制或二次幂方面考虑一下。
首先,我们知道当n=a+b时, xn=xa∗xb ,另一方面,任何一个正数都可以用2次幂和的形式来表示,如26 = 0b11010(二进制) = 16+8+2,则 x26=x16∗x8∗x2 ,这样的话,我们只需考虑 x2i(i=0,1,2,3...) ,通过把n表示成2次幂和的形式,按照公式相乘即可。根据这个原理,代码如下:
class Solution {
public:
double myPow(double x, int n) {
// 一些特殊情况,直接进行处理
if (n == 0) return 1.0;
if (x == 0) return 0.0;
if (x == 1.0) return 1.0;
// 把n当做整数计算
unsigned int nn = abs(n);
double tmp = x; // tmp 用来存储x^(2^i),初值i=0,tmp=x
double res = 1.0;
// 循环查看nn的最后一位,
while (nn != 0){
if((nn & 1) == 1) res *= tmp; // 最后一位为1,将结果乘上tmp
if (res == 0 && n > 0) return 0.0; // res为0后,就没必要再计算了
tmp *= tmp; // 更新tmp,tmp = x -> x^2 -> x^4 -> x^8 ...
nn >>= 1; // 右移一位
}
return n > 0 ? res : 1.0/res; // n为负数时,直接返回1.0/res
}
};