數論算法模板總結

公約數

GCD

LL GCD( LL a,LL b ) 
{
    return b==0?a:GCD(b,a%b);
}

 EX_GCD

LL EX_GCD( LL a,LL b,LL &x,LL &y )//ax+by=gcd(a,b)
{
    LL d=a;
    if( !b ) x=1;y=0;
    else {
        d=EX_GCD(b,a%b,y,x);
        y-=a/b*x;
    }
    return d;//返回最大公因數
}
/*  
    求a * x + b * y = c的整數解。
    一、先計算Gcd(a,b)，若n不能被Gcd(a,b)整除，則方程無整數解；不然，在方程兩邊同時除以Gcd(a,b)，獲得新的不定方程a' * x + b' * y = c'，此時Gcd(a',b')=1;
    二、利用上面所說的歐幾里德算法求出方程a' * x + b' * y = 1的一組整數解x0,y0，則c' * x0,c' * y0是方程a' * x + b' * y = c'的一組整數解；
    三、根據數論中的相關定理，可得方程a' * x + b' * y = c'的全部整數解爲：
    x = c' * x0 + b' * t
    y = c' * y0 - a' * t
    (t爲整數)
*/

 

素數

素數的三種篩法

樸素算法   //O( n*sqrt(n) )

bool Isprime( LL n )
{
    for( LL i=2;i*i<=n;i++ )
        if( n%i==0 ) return false;
    return true;
}

Eratosthenes篩法   //O( n*log n )

void Eratosthenes( int n )
{
    memset(Isprime,true,sizeof(Isprime));
    for( int i=2;i<=n;i++ )
    {
        if( Isprime(i) )
            for( int j=i*i;j<=n;j+=i )
                Isprime[j]=false;
    }
}

歐拉算法   //O(n)

void Euler(int n)
{
    memset(Isprime,0,sizeof(Isprime));
    for( int i=2;i<=n;i++ )
    {
        if( ！Isprime[i] )
            prime[cnt++]=i;
        for( int j=0;j<cnt;j++ )
        {
            if( prime[j]*i>n ) break;
            Isprime[ prime[j]*i ] =true;  //與下一行代碼不可交換
            if( i%prime[j]==0 ) break;
        }
    }
}

 

冪運算

 快速冪  //O(log n)

LL Pow( LL x,LL n )
{
    LL res=1;
    while(n){
        if( n&1 ) res*=x;
        x*=x;
        n>>=1;
    }
    return res;
}