백준 2740번 행렬 곱셈

문제 풀이

• 행렬의 곱셈을 그대로 구현하면 되는 문제

코드 풀이

import sys
input = sys.stdin.readline

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)]

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

for n in range(N):
    for k in range(K):
        for m in range(M):
            result[n][k] += A[n][m] * B[m][k]

for i in result:
    print(*i)