1. 程式人生 > >快速冪與快速矩陣冪(以大數下的斐波那契數列為例)

快速冪與快速矩陣冪(以大數下的斐波那契數列為例)

一般地,a^n的演算法時間複雜度為o(n),但是如果n為大數,則執行時間過長,效率不高。因此,使用二分的思想降低時間複雜度,使其降至o(logn),則會使執行效率較大提升。二分思想如下圖所示。

例如:2^8=2^4*2^4=(2^2*2^2)*(2^2*2^2)=((2*2)*(2*2))*((2*2)*(2*2)),只需計算4次,比原來的8次降低許多(這裡多一次,是把第一次乘1也算上)

  奇數次:2^9=2*(2^4)*(2^4)=2*(2^2*2^2)*(2^2*2^2)=2*((2*2)*(2*2))*((2*2)*(2*2)) 

因此,快速冪的程式碼實現如下:

def quick(a,n,p):
    res=1
    ans=a
    while n>=2:
        if n&1:
            res=(ans*res)%p
        ans=(ans*ans)%p
        n=n>>1
    ans=res*ans
    return ans%p
if __name__ == '__main__':
    a,n,p=map(int,raw_input().split())
    print quick(a,n,p)

借鑑快速冪的思想,求快速矩陣冪。

根據矩陣乘法的結合律AB.....B=A(B)^n,而B^n則可以用快速冪的方法求,就可以降低計算時間。

矩陣的快速冪是用來高效地計算矩陣的高次方的。將樸素的o(n)的時間複雜度,降到log(n)。這裡先對原理(主要運用了矩陣乘法的結合律)做下簡單形象的介紹:一般一個矩陣的n次方,我們會通過連乘n-1次來得到它的n次冪。但做下簡單的改進就能減少連乘的次數,方法如下:把n個矩陣進行兩兩分組,比如:A*A*A*A*A*A  =>  (A*A)*(A*A)*(A*A)這樣變的好處是,你只需要計算一次A*A,然後將結果(A*A)連乘自己兩次就能得到A^6,即(A*A)^3=A^6。算一下發現這次一共乘了3次,少於原來的5次。其實大家還可以取A^3作為一個基本單位。原理都一樣:利用矩陣乘法的結合律,來減少重複計算的次數。以上都是取一個具體的數來作為最小單位的長度,這樣做雖然能夠改進效率,但缺陷也是很明顯的,取個極限的例子(可能有點不恰當,但基本能說明問題),當n無窮大的時候,你現在所取的長度其實和1沒什麼區別。所以就需要我們找到一種與n增長速度”相適應“的”單位長度“,那這個長度到底怎麼去取呢???這點是我們要思考的問題。有了以上的知識,我們現在再來看看,到底怎麼迅速地求得矩陣的N次冪。既然要減少重複計算,那麼就要充分利用現有的計算結果咯!~怎麼充分利用計算結果呢???這裡考慮二分的思想。。
回頭看看矩陣的快速冪問題,我們是不是也能把它離散化呢?比如A^19  =>  (A^16)*(A^2)*(A^1),顯然採取這樣的方式計算時因子數將是log(n)級別的(原來的因子數是n),不僅這樣,因子間也是存在某種聯絡的,比如A^4能通過(A^2)*(A^2)得到,A^8又能通過(A^4)*(A^4)得到,這點也充分利用了現有的結果作為有利條件。下面舉個例子進行說明:

斐波那契數列的遞推: f(n) = f(n-1) + f(n-2);其對應的矩陣就是      f(n)   f(n-1)    f(n-1)   f(n-2)     1    1          即形如C=A*B
                                                                                                                      0      0   =        0       0    *   1    0    
可以發現,左面是個常數矩陣,而右面的列向量中每一項就是遞推公式中的依賴項。然後根據初始兩f(1)=1,f(0)=0,則將初始值代入A中,求C=A*(B^(n-2))。

問題描述:

現在有一棟高樓,但是電梯卻出了故障,無奈的你只能走樓梯上樓,根據你的腿長,你一次能走1級或2級樓梯,已知你要走n級樓梯才能走到你的目的樓層,請計算你走到目的樓層的方案數,由於樓很高,所以n的範圍為int範圍內的正整數。
給定樓梯總數n,請返回方案數。為了防止溢位,請返回結果Mod 1000000007的值。
測試樣例:
      3
返回:3

程式碼實現:

#-*- coding:utf-8 -*-
class GoUpstairs:
    def countWays(self, n):
        n=n-1
        res=base=[[1,1],[1,0]]
        surplus=[[1,1],[0,0]]
        while(n>=2):
            if n&1:
                surplus=self.mutiply(surplus,res)
            res=self.mutiply(res,res)
            n=n>>1
        res=self.mutiply(surplus, res)
        return res[0][0]%1000000007
    def mutiply(self,a,b):
        temp=[[0,0],[0,0]]
        for i in range(len(a)):
            for j in range(len(b)):
                for k in range(len(temp)):
                    temp[i][j]+=a[i][k]*b[k][j]%1000000007
                      
        return temp

ps:遇到變形的遞推公式,也可以用矩陣快速冪,只需找對相應的矩陣關係即可