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

문제 링크

---

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

 

풀이 코드

---

import sys
input = sys.stdin.readline

n, m = map(int,input().split())
a = [list(map(int,input().split())) for i in range(n)]

m, k = map(int,input().split())
b = [list(map(int,input().split())) for i in range(m)]

for i in range(n):
    result = []
    for j in range(k):
        x = 0
        for l in range(m):
            x += a[i][l] * b[l][j]
        result.append(x)
        
    print(*result)

 

코드 해석

---

행렬곱을 구현한다.

 

체감 난이도: ★★☆☆☆