백준 2740번 "행렬 곱셈"

문제

백준 2740번 행렬곱셈

풀이

이미지 출처 : 위키백과

3중 반복문을 이용하여 행렬곱셈을 처리하면 된다.

Python 코드

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
A = []
for _ in range(n):
    A.append(list(map(int, input().split())))

m, k = map(int, input().split())
B = []
for _ in range(m):
    B.append(list(map(int, input().split())))


#행렬 곱셈
C = [[0 for _ in range(k)] for _ in range(n)]

for i in range(n):
    for j in range(k):
        for l in range(m):
            C[i][j] += A[i][l] * B[l][j]

#출력문
for i in C:
    for j in i:
        print(j, end = ' ')
    print('')