1. 程式人生 > >數論求逆元的三種方法

數論求逆元的三種方法

擴充套件歐幾里德演算法

//非遞迴的擴充套件歐幾里德演算法  
//返回a、b的gcd,同時x、y滿足ax+by=gcd  
int_t exEuclid(int_t a,int_t b,int_t&x,int_t&y){  
    int_t x0 = 1, y0 = 0;  
    int_t x1 = 0, y1 = 1;  
    x = 0; y = 1;  
    int_t r = a % b;  
    int_t q = ( a - r ) / b;  
    while( r ){  
        x = x0 - q * x1;  
        y = y0 - q * y1;  
        x0 = x1; y0 = y1;  
        x1 = x; y1 = y;  
        a = b; b = r; r = a % b;  
        q = ( a - r ) / b;  
    }  
    return b;  
}  
//求a相對於p的逆元,a、p互質才存在逆元  
int_t inv(int_t a,int_t p){  
    int_t x,y;  
    int_t r = exEuclid(a,p,x,y);  
    if ( r != 1 ) return 0;  
    x = x % p;  
    if ( x < 0 ) x += p;  
    return x;  
}  


費馬小定理

//計算a^b%mod
llt powerMod(llt a,llt b,llt mod){
    llt ret = 1LL;
    a %= mod;
    while( b ){
        if ( b & 1LL ) ret = (ret*a)%mod,--b;
        b >>= 1LL;
        a = multiMod(a,a,mod);
    }
    return ret;
}
llt inv( int x ,int mod ){
    return powerMod( x, mod-2 ,mod );
}




線性篩逆元

這個做法實際上是這樣的,首先 111(modp)1−1≡1(modp)

然後我們設 p=ki+r,r<i,1<i<pp=k⋅i+r, r<i, 1<i<p

再將這個式子放到modpmodp 意義下就會得到

ki+r0(modp)k⋅i+r≡0(modp)

兩邊同時乘上 i1r1i−1⋅r−1 就會得到

kr1+i1i1i10kr1pi(pmodi)1(modp)(modp)(modp)k⋅r−1+i−1≡0(modp)i−1≡−k⋅r−1(modp)i−1≡−⌊pi⌋⋅(pmodi)−1(modp) 

於是就可以從前面推出當前的逆元了,程式碼也就一行


int const  Ns = 3e4+7;
llt inv[Ns]={0,1};
void init(){
    for (int i = 2;i <= Ns;++i ){
        inv[i]=inv[MOD%i]*(MOD-MOD/i)%MOD;
    }
}


參考