[백준/파이썬] 2740번 행렬 곱셈

-코드

from sys import stdin
n, m = map(int, stdin.readline().split())
arr = []
arr2 = []
for _ in range(n):
    nums = list(map(int, stdin.readline().split()))
    arr.append(nums)
m, k = map(int, stdin.readline().split())
for _ in range(m):
    nums2 = list(map(int, stdin.readline().split()))
    arr2.append(nums2)
answer = [[0 for _ in range(k)] for _ in range(n)]
for i in range(n):
    for j in range(k):
        for l in range(m):
            answer[i][j] += arr[i][l] * arr2[l][j]
for i in answer:
    print(*i)

처음에 answer 리스트를 안 만들고 해결을 하려고 하다가 막혀서 고민하다 다시 만들어서 해결을 하였습니다.