費馬小定理求逆元
阿新 • • 發佈:2018-12-24
求餘的概念:
(a + b) % p = (a%p + b%p) %p
(a - b) % p = (a%p - b%p) %p
(a * b) % p = (a%p * b%p) %p
為什麼要求逆元:對於一些題目,我們必須在中間過程中進行求餘,否則數字太大,電腦存不下,那如果這個算式中出現除法,我們是不是對這個算式就無法計算了呢?這時候就用到了逆元。
費馬曾經說過:
費馬小定理
a^(p-1) ≡1 (mod p)
兩邊同除以a
a^(p-2) ≡1/a (mod p)
應該寫a^(p-2) ≡ inv(a) (mod p)
所以inv(a) = a^(p-2) (mod p)
這個用快速冪求一下,複雜度O(logn)
程式碼:
LL pow_mod(LL a, LL b, LL p){//a的b次方求餘p
LL ret = 1;
while(b){
if(b & 1) ret = (ret * a) % p;
a = (a * a) % p;//乘法還可以稍微優化一下
b >>= 1;
}
return ret;
}
LL Fermat(LL a, LL p){//費馬求a關於b的逆元
return pow_mod(a, p-2, p);
}