分治算法——快速幂
(我貌似不会写二进制拆分的快速幂。。。。)
对于求解a^b mod p,我们的暴力手段是模拟,将b个a一个一个乘上去再取模
时间显然是O(b),当b有几千万,乃至几亿时,我们发现这显然会超时
思考有没有优化的空间呢
我们发现求解a^b mod p 等价于下面的问题:
(a^2) ^ (b/2) mod p (当b为偶数时)
(a^2) ^ (b div 2) * a mod p (当b为奇数时)
我们显然可以将这个求解 a^b mod p这个原问题,转化为上述与原问题结构相同,但是规模更小的子问题来解决(这显然是一个分治)
我们不妨用ans表示问题的答案,同时当问题分离出一个a时,我们就直接将其乘入ans中
通过不断的递归,当b为0时可以直接出解了,由于每一次对于原问题将其分解都会使得b/2,则会发现其时间复杂度为O(lg b)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
var b,p,k,ans,x,y,z:longint; begin readln(b,p,k); x:=b; y:=p; z:=k; ans:=1; b:=b mod k; while p<>0 do begin if p and 1=1 then ans:=ans*b mod k //p是否为奇数 else ans:=ans mod k; b:=(b*b) mod k; //每一次分解都会使得b变为b^2 p:=p>>1; //p减少了一般 end; writeln(x, '^' ,y, ' mod ' ,z, '=' ,ans mod k); end.、 //不写递归的原因是递归比较慢 |