经常会遇到对一个数进行幂运算,如求
x
n
x^n
xn,朴素的算法是把这个数乘n次
,复杂度为O(n),在n比较大的情况就会非常慢。而快速幂可以在O(logn)的复杂度内计算乘方。
快速幂用到的是二进制的思想。比如要算出
5
31
5^{31}
531,我们把31转换为2进制,可以转化为
5
(
11111
)
2
=
5
2
4
⋅
5
2
3
⋅
5
2
2
⋅
5
2
1
⋅
5
2
0
5^{(11111)_2}=5^{2^4}·5^{2^3}·5^{2^2}·5^{2^1}·5^{2^0}
5(11111)2=524⋅523⋅522⋅521⋅520,而
x
2
i
+
1
=
(
x
2
i
)
2
x^{2^{i+1}}=(x^{2^i})^2
x2i+1=(x2i)2,于是我们只需要不断把底数取平方再把各项乘起来就好了。
代码如下
int qpow(int x, int n){
int ans = 1;
while(n){
if(n & 1) ans *= x;//位运算,判断n的末位是否为1
//如果满足则让ans乘当前的x底数
x *= x ;//底数取平方
n >>= 1;//让n右移一位比如1010右移变成了0101相当于整除以二
}
return ans;
}
我们可以根据代码推敲一下这个过程,例如
5
29
5^{29}
529
5
(
29
)
10
=
5
(
11101
)
2
5^{(29)_{10}} = 5^{(11101)_2}
5(29)10=5(11101)2
传入函数后
n = 11101, x = 5,
① n & 1 = 1, ans = ans * x = 1 * 5 = 5, x = x * x =
5
2
5^2
52 , n = n>>1 = 1110
②n & 1 = 0, x = x * x =
5
4
5^4
54, n = n>>1 = 111
③n & 1 = 1, ans = ans * x =
5
∗
5
4
=
5
(
101
)
2
5*5^4=5^{(101)_2}
5∗54=5(101)2, x = x * x =
5
8
5^8
58, n = n>>1 = 11;
④n & 1 = 1, ans = ans * x =
5
(
101
)
2
∗
5
8
=
5
(
1101
)
2
5^{(101)_2}*5^8=5^{(1101)_2}
5(101)2∗58=5(1101)2, x = x * x =
5
16
5^{16}
516, n = n>>1 = 1;
⑤n & 1 = 1, ans = ans * x =
5
(
1101
)
2
∗
5
16
=
5
(
11101
)
2
=
5
29
5^{(1101)_2}*5^{16}=5^{(11101)_2}=5^{29}
5(1101)2∗516=5(11101)2=529, x = x * x =
5
32
5^{32}
532, n = n>>1 = 0
n = 0,跳出while循环,返回答案ans =
5
29
5^{29}
529。
由于n经常很大,涉及取模,那么我们的模板可以改为
long long qpow(long long x, long long n){
long long ans = 1;
while(n){
if(n & 1) ans = (x * ans) % p;
x = (x * x) % p;
n >>= 1;
}
return ans % p;
}
其中p为模
此外,快速幂也可以用在对矩阵进行幂运算上。只要底数满足乘法和乘法结合律,快速幂应该都是有效的。