python矩阵乘法代码,[蓝桥杯][基础练习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()