问题:计算x的n次幂,该结果对Mod进行取模。
注意:由于该结果的值可能会很大,需考虑可能出现TLE 等情况。
def func(x, n, Mod):
res = 1
x %= Mod
while n != 0:
if n & 1: #取n的2进制的最低位
res = (res * x) % Mod
n >>= 1 #相当于n//2
x = (x * x) % Mod
return res
数学原理:(a * b) % Mod = [(a % Mod) * (b % Mod)] % Mod
设 a = c * Mod + d, b = e * Mod + f。
则(a * b) % Mod
= ((c * Mod + d) * (e * Mod + f)) % Mod
= (c * e * Mod ** 2 + (c * f + d * e) * Mod + d * f) % Mod
= (d * f) % Mod
且((a % Mod) * (b % Mod)) % Mod
= (((c * Mod + d) % Mod) * ((e * Mod + f) % Mod)) % Mod
= (d * f) % Mod
故(a * b) % Mod = [(a % Mod) * (b % Mod)] % Mod = (d * f) % Mod