1. 程式人生 > >3的冪的和(快速冪)

3的冪的和(快速冪)

求:3^0 + 3^1 +...+ 3^(N) mod 1000000007

Input
輸入一個數N(0 <= N <= 10^9)
Output
輸出:計算結果
Sample Input

3

Sample Output

40

要用到的知識:

**逆元:
若對於數字A,C 存在X,使A * X = 1 (mod C) ,那麼稱X為 A 對C的乘法逆元。
理論依據:
F / A mod C = ?
如果存在 A*X = 1 (mod C)
那麼2邊同時乘起來,得到 F * X = ? (mod C)**

成立條件

(1) 模方程 A * X = 1(mod C) 存在解

(2) A | F (F % A == 0)

若ax=1 mod f 則稱a關於模f的乘法逆元為x。也可表示為ax≡1(mod f)。

  當a與f互素時,a關於模f的乘法逆元有唯一解。如果不互素,則無解。如果f為素數,則從1到f-1的任意數都與f互素,即在1到f-1之間都恰好有一個關於模f的乘法逆元
a*b

快速乘法的基本思想 ,是二進位制和乘法分配律的結合,(不由得想起浮點數不滿足結合律,嚴重吐槽!!!╮(╯-╰)╭),比如說,13 ==(1101)2 ,4*13等於4*(1101)2 ,用分配律展開得到4*13 == 4*(1000+100+1)2,我們不難觀察出,快速冪可以通過判斷當前的位(bit)是1還是0,推斷出是否需要做求和操作,每次移動到下一位(bit)時,就對ans進行*2操作,等待是否求和。由於除以2和位移操作是等效的,因此這也可以看作是二分思想的應用,這種演算法將b進行二分從而減少了不必要的運算,時間複雜度是log(n)。

a^b

快速冪其實可以看作是快速乘法的特例,在快速冪中,我們不再對ans進行2操作,因為在a^b中b的意義已經從乘數變成了指數,但是我們可以仍然把b寫成二進位制,舉例說明:此時,我們將4*13改為4^13,13=(1101)2 ,二進位制13寫開我們得到(1000+100+1),注意,這裡的所有二進位制是指數,指數的相加意味著底數相乘,因此有4^13 == 48 * 44 * 41。再注意到指數之間的2倍關係,我們就可以用很少的幾個變數,完成這一演算法。這樣,我們就將原本用迴圈需要O(n)的演算法,改進為O(logN)的演算法。*

按照慣例,給出儘可能簡潔高效的程式碼實現 (以下所有int都可用long long 代替)

首先,給出快速乘法的實現:

1 //快速乘法 
2 int qmul(int a,int b){// 根據資料範圍可選擇long long 
3     int ans=0;
4     while(b){
5         if( b&1)ans+=a;//按位與完成位數為1的判斷
6         b>>=1;a<<=1;//位運算代替/2和*2
7     }
8     return ans;
9 }

如果涉及到快速乘法取模,則需要進行一些微小改動

1 //快速乘法取模 
2 int qmul_mod(int a,int b,int mod){
3     int ans=0;
4     while(b){
5         if((b%=mod)&1)ans+=a%=mod;//這裡需要b%=mod 以及a%=mod 
6         b>>=1;a<<=1;
7     }
8     return ans%mod;  //ans也需要對mod取模 
9 }

接下來是快速冪的實現:

1 //快速冪 a^b 
 2 int qpow(int a,int b){
 3     if(a==0)return 0;//這是個坑,校賽被坑過,很多網上的實現都沒寫這一點
 4     int ans=1;
 5     while(b){
 6         if(b&1)ans*=a;//和快速乘法的區別
 7         b>>=1;a*=a;//區別,同上
 8     }
 9     return ans;
10 } 

以及含有取模的快速冪:


int qpow_mod(int a,int b,int mod){
    if(a==0)return 0;
    int ans=1;
    while(b){
        if(b&1)ans=(ans%mod)*(a%mod);//如果確定資料不會爆的話,可寫成 ans*=a%=mod;
        b>>=1;a*=a%=mod;//等價於a=(a%mod)*(a%mod),且將一個模運算通過賦值代替,提高了效率
    }
    return ans%mod;//資料不會爆的話,這裡的%運算會等價於第5中不斷重複的 ans%mod
}

如果我們對於效能還有更進一步的要求,那麼也就是減少取模運算了,那麼我們需要確定資料範圍不會爆掉

在這樣的前提下,我們可以只用原先1/4的取模運算量完成快速冪

int qpow_mod(int a,int b,int mod){
    if(!a)return 0;
    int ans=1;
    while(b){
        if(b&1)ans*=a%=mod;//這裡的模運算只有一個
        b>>=1;a*=a;//這裡的模運算沒有了
    }
    return ans%mod;
}

完整程式碼:

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define MOD 1000000007
typedef long long LL;
LL quick_mul(LL x,LL n){
    LL res=1;
    while(n){
    if(n&1)
        res=(res*x)%MOD;
    x=x*x%MOD;
    n/=2;
    }
    return res;
}
int main()
{
    LL n,sum;
    scanf("%I64d",&n);
    sum=(quick_mul(3,n+1)-1)*500000004%MOD;//運用到了等比數列求和公式,等價於(3^(n+1)-1)/2,2對1000000007的逆元為500000004。
    printf("%I64d",sum);
    return 0;
}