快速幂
当遇到题目要求输出返回值要%1E9+7的时候,基本上,该题最终可以写成ab的形式,可以用“快速幂”方法在O(logn)的时间复杂度求解ab。
核心思想:
b可以展开为二进制,以a3次方为例,3 = 0b011。
a3 = (a1)1 * (a2)1 * (a4)0
发现:a1 → a2 → (a2)2→ …,可以通过分治法,不断迭代求得
当b的某位为1时, 它将会纳入最终结果的运算。
上述方法可以用位运算实现。
取模运算的时候,同理,只要注意模运算的公式:
(1) ab % c = ( a % c) b % c
(2) (a * b) % c = (a % c * b % c) % c
代码模板:
def power(a, b): # a^b
res = 1
while b > 0:
if b & 1 == 1:
res *= a
a = a * a
b >>= 1
return res
def powerMod(a, b, c): # (a^b)%c
res = 1
while b > 0:
if b & 1 == 1:
res = (res * a) % c
a = (a * a) % c
b >>= 1
return res