快速幂: 快速计算底数的n次幂
时间复杂度: O(log₂N)
要计算乘幂,我们一般就直接这样写:
double myPow(double x, int n) {
double ans=1;
int flag = 0;
if(n<0) {n=-n; flag=1;}
for(int i=1;i<=n;i++)
ans*=x;
if(flag) ans=1/ans;
return ans;
}
但这种做法的时间复杂度为O(N),当N很大时,效率并不理想。
此时我们可以把n拆解为
2
k
1
+
2
k
2
+
.
.
.
+
2
k
m
2^{k_1} + 2^{k_2} + ...+2^{k_m}
2k1+2k2+...+2km m项之和
只需转化为二进制然后逐一获取每一位即可得到
k
1
,
k
2
,
k
3
,
.
.
.
,
k
m
k_1,k_2,k_3,...,k_m
k1,k2,k3,...,km
x
n
=
x
2
k
1
+
2
k
2
+
.
.
.
=
x
2
k
1
x
2
k
2
.
.
.
x
2
k
m
.
x^n=x^{2^{k1}+2^{k2}+...}=x^{2^{k_1}}x^{2^{k_2}}...~x^{2^{k_m}}.
xn=x2k1+2k2+...=x2k1x2k2... x2km.
double myPow(double x, int n) {
double base = x;
double ans = 1;
int flag=0;
if(n<0) {flag=1; n=-n;} //(*)
while (n != 0)
{
if (n & 1)
{
ans *= base; //1
}
base *= base; //2
n >>= 1;
}
if(flag) ans=1/ans;
return ans;
}
base用来存储每一个形如 x 2 i x^{2^i} x2i的因子
-
注释1:n&1 不为0 说明n的最后一位(因为后面的n>>=1一直在移除最后一位,因此这个最后一位可以表示第i位, i 从0->m-1)为1,那么就有因子 x 2 i x^{2^i} x2i 比如一开始
base=x
,如果最低位(第0位)为1,那么此时a n s = 1 ∗ x 2 0 = x ans=1*x^{2^0} =x ans=1∗x20=x
-
注释2:
第1位base就由第0位的base=x
通过base*=base
变成 x 2 = x 2 1 x^2=x^{2^1} x2=x21, 第2位base通过base*=base,就变成 x 2 ∗ x 2 = x 4 = x 2 2 x^2 * x^2 = x^4 = x^{2^2} x2∗x2=x4=x22,依次类推
需要注意的是,在
(
∗
)
(*)
(∗)号处如果
n
=
−
2
31
n=-2^{31}
n=−231 那么n=-n 这步会溢出
因为int的范围是
−
2
31
到
2
31
−
1
-2^{31} 到~ 2^{31}-1
−231到 231−1 , 如果数据规模至此则考虑换用long long来存储