[PS] 백준 2740번 - 행렬 곱셈 | Python

백준 2740번 문제 write-up입니다.

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

2740번: 행렬 곱셈

---

행렬 곱셈 알고리즘을 작성하는 문제입니다.

 

다른 조건 없이 행렬 곱셈 알고리즘을 코드로 구현하면 끝나는 문제이기 때문에, 풀이는 아래에 link된 게시물을 참고해주세요.

https://spareone.tistory.com/72

[Algorithm] C++로 구현한 행렬 곱셈 알고리즘

---

an, am = map(int, input().split())

a = []
for i in range(an):
    temp = list(map(int, input().split()))
    a.append(temp)


bm, bk = map(int, input().split())

b = []
for i in range(bm):
    temp = list(map(int, input().split()))
    b.append(temp)


ans = []
for i in range(an):
    res = []
    for j in range(bk):
        temp = 0
        for k in range(am):
            temp += a[i][k] * b[k][j]
        res.append(temp)
    ans.append(res)

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