[백준/BOJ/알고리즘/파이썬(python)]#2740_행렬 곱셈

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

2740번: 행렬 곱셈

행렬의 곱셈을 계산하는 문제로

반복문과 배열을 행렬곱연산에 맞게 구조를 잘 짜면 된다.

 

파이썬 코드

#2740_행렬곱셈
import sys
input = sys.stdin.readline

N, M = map(int, input().split())
A_element = []
for _ in range(N):
    A_element.append(list(map(int,input().split())))

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

tmp_el=0
tmp=[]
res = []
for i in range(N):
    for j in range(K):
        for a in range(M):
            tmp_el += A_element[i][a]*B_element[a][j]
        tmp.append(tmp_el)
        tmp_el = 0
    res.append(tmp)
    tmp = []

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