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

11 Apr 2021 in Dev on BOJ

2740 행렬 곱셈 바로가기

접근방식

NM 행렬과 MK 행렬이 만나 N*K행렬이 만들어진다는 것을 기억하면 된다!

파이썬 코드

import sys

N, M = map(int, sys.stdin.readline().split())
A = []
for _ in range(N):
    A.append(list(map(int, sys.stdin.readline().split())))

M, K = map(int, sys.stdin.readline().split())
B = []
for _ in range(M):
    B.append(list(map(int, sys.stdin.readline().split())))


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

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

for i in C:
    for j in i:
        print(j, end = ' ')
    print()

참고

https://claude-u.tistory.com/262