请在O(logn)时间内求a的n次幂。
1.递归版
看到logn,自然想到二分。
比如求3的12次幂,我们用二分可以写成
((32)2 *3)2.
O(n)复杂度要乘以n次,这样可以乘以4次。
- 32 一次
- (32)2 一次
- (32)2*3 一次
- ((32)2*3)2 一次
递归函数:
2.非递归版
将幂写成2进制形式。
当二进制位为0的时候我们不用算,因为就是1.
当二进制位为1的时候我们把它乘入即可。
因为n约有logn+1个二进制位,所以可以进行logn次乘法,复杂度也满足。
剩下的就是如何乘了,知道
所以
int ans=1;
while(n!=0){
if(n&1==1)//与运算,判断最后一位是不是1
{
ans=ans*a;//计入答案
}
a*=a;//a=a*a,计算下一个
n>>=1;//移位,判断当前位为1或0
}
return ans;
二分
for(int i = n; i != 0; i /= 2){
if(i % 2 != 0){
ans *= a;
}
a*= a;
}
return n < 0 ? 1 / ans: ans;
参考 https://oi-wiki.org/math/quick-pow/#_4