蓝桥杯基础练习VIP-矩阵乘法---矩阵快速幂（python）

解题思路:
    使用快速幂算法；

    
注意事项:
    将原来快速幂算法中的乘法换成矩阵乘法，因此我们要先定义矩阵乘法。

参考代码:

n,m = map(int,input().split())
A = [list(map(int,input().split())) for i in range(n)]

# 定义矩阵乘法
def mult_matrix(M,N):
    R = [[0]*len(N[0]) for i in range(len(M))]
    
    for a in range(len(M)):
        for b in range(len(N[0])):
            ans = 0
            for i in range(len(M[a])):
                ans += M[a][i]*N[i][b]
            R[a][b] = ans
    return R
    
# 定义矩阵快速幂
def quickpow(base, power):
    # 单位阵
    res = [[0]*len(base[0]) for i in range(len(base))]
    for i in range(len(base)):
        res[i][i] = 1
        
    # 快速幂算法
    while power>0:
        if power & 1:
            res = mult_matrix(res,base)
        power >>= 1
        base = mult_matrix(base,base)
    
    return res

res = quickpow(A,m)
for a in res:
    for b in a:
        print(b,end=' ')
    print()