蓝桥杯基础练习VIP-矩阵乘法-题解（Python代码）

解题思路:
        不知为何总是报输出超限43，思路应该是没问题的，有小伙伴晓得为啥的评论下让我看到谢谢。

参考代码:

n,m = map(int,input().split())
nums = [list(map(int,input().split())) for _ in range(n)]
res = [[0]*n for _ in range(n)]
k = 0
i,j = 0,0
for t in range(1,m):
    for i in range(n):
        for k in range(0,n):
            r = 0
            for j in range(n):
                r += nums[i][j]*nums[j][k]
            res[i][k] = r
    for i in range(n):
        for j in range(n):
            nums[i][j] = res[i][j]
if m==0:
    for i in range(n):
        for j in range(n):
            if i==j:
                print("1",end=" ")
            else:
                print("0",end=" ")
        if i!=(n-1):
            print()
else:
    for i in range(n):
        for j in range(n):
            print(nums[i][j],end=" ")
        if i!=(n-1):
            print()