Python 不使用numpy，求自定义矩阵的n次幂 / 蓝桥杯 基础练习 vip题库 — 矩阵乘法

题目

重点代码

def matrix_2mi(a, b):
    J = range(N)
    K = range(N)
    I = range(N)
    return [[sum(a[i][j] * b[j][k] for j in J) for k in K] for i in I]

完整代码

def matrix_2mi(a, b):
    J = range(N)
    K = range(N)
    I = range(N)
    return [[sum(a[i][j] * b[j][k] for j in J) for k in K] for i in I]


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

matrix = []
for i in range(N):
    juzhen = list(map(int, input().split()))
    matrix.append(juzhen)

out_matrix = []

if M == 0:
    for i in range(N):
        out_matrix.append([0] * N)
    for i in range(N):
        out_matrix[i][i] = 1
else:
    out_matrix = matrix
    for i in range(M-1):  #2次幂只循环1次
        out_matrix = matrix_2mi(out_matrix, matrix)

#print(out_matrix)
for i in range(N):
    for j in range(N):
        print(out_matrix[i][j], end = '')
        if j != (N - 1):
            print(end = ' ')
    print(end = '\n')

评测结果

参考

1. How to multiply two matrices together in Python [duplicate]
2. Python Matrix Multiplication Variations