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

 

문제 : https://www.acmicpc.net/problem/2740

 

 

코드

N, M = map(int, input().split())
matA = []
matB = []

for _ in range(N):
    matA.append(list(map(int, input().split())))

M, K = map(int, input().split())
for _ in range(M):
    matB.append(list(map(int, input().split())))

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

for n in range(N):
    for k in range(K):
        for m in range(M):
            matrix[n][k] += matA[n][m] * matB[m][k]

for mat in matrix:
    for j in mat:
        print(j, end = " ")
    print()

 

알아둘 것

행렬의 곱셈은 N*M 행렬과 M*K 행렬이 만나 N*K행렬을 만든다