백준 2740 - 행렬 곱셈 (파이썬)

Answer

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

a = [list(map(int, input().split())) for _ in range(N)]

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

b = [list(map(int, input().split())) for _ in range(M)]

answer = [[0 for _ in range(K)] for _ in range(N)]

for i in range(N):
    for j in range(K):
        for k in range(M):
            answer[i][j] += a[i][k] * b[k][j]

for i in answer:
    print(*i)