快速幂&矩阵快速幂&线性递推转化为矩阵快速幂

##二进制拆分求快速幂

b,p,k=map(int,input().split())

def ksm(a,b,c):

    ans=1%c

    while b:

        if b%2==1: ## if b&1:

            ans=ans*a%c

            ans%=c

        b//=2 ## b>>=1

        a=a*a%c

    return ans

print(ksm(b,p,k))

矩阵快速幂

def read(n):

    ##读入一个n行的矩阵

    a=[]

    for i in range (n):

        a.append(list(map(int,input().split())))

    return a

def output(A):

    ##打印一个矩阵

    n=len(A)

    for i in range (n):

        print(" ".join(map(str,A[i])))

def mul(A,B):

    ##c[i][j]+=a[i][k]*b[k][j]

    ##矩阵A与矩阵B相乘递推公式

    m,n=len(A),len(A[0])

    n_,k=len(B),len(B[0])

    C=[[0]*m for i in range (k)]

    if n!=n_:

        return None

    for i in range (m):

        for j in range (k):

            for r in range (n):

                C[i][j]+=A[i][r]*B[r][j]

    return C

def ksm(A,b):##求矩阵A的b次方

    n=len(A)

    ans=[[1 if i==j else 0 for i in range (n)]for j in range (n)]##单位矩阵

    while b:

        if b&1:

            ans=mul(ans,A)

        b>>=1

        A=mul(A,A)

    return ans

N,M=map(int,input().split())

output(ksm(read(N),M))

线性递推转化为矩阵快速幂

例题:斐波那契数列 用矩阵快速幂递推

F[0]=0 F[1]=1

F[n]=1*F[n-1]+1*F[n-2]    (1)

F[n-1]=1*F[n-1]+0*F[n-2] (2)

(1)(2)式线性方程组得到矩阵递推式:

  • 3
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值