1. 程式人生 > >費馬小定理求逆元

費馬小定理求逆元

求餘的概念:

(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);
}