백준 2740 풀이 (행렬 곱셈)

이 문제는 어렵지는 않았지만 살짝 귀찮은 문제였다. 나는 당연히 무식하게 for 돌려서 풀었다.

import sys
n,m = map(int, sys.stdin.readline().split())
a = []
for _ in range(n):
    a.append(list(map(int,sys.stdin.readline().split())))
n2,m2 = map(int, sys.stdin.readline().split())
b=[]
for _ in range(n2):
    b.append(list(map(int,sys.stdin.readline().split())))
final = []
for i in range(n):
    t = a[i]
    total = []
    for j in range(m2):
        tmp = []
        t = 0
        for row in range(n2):
            tmp.append(b[row][j])
        for n in range(n2):
            t+= a[i][n] * tmp[n]
        total.append(t)
    final.append(total)
for i in final:
    for j in i:
        print(j,end=' ')
    print()

답은 맞았지만 문제의 의도가 이렇게 푸는게 맞는지 다른 고수의 코드를 구경하러 갔다.

그랬더니 미리 배열을 구해서 풀었더니 훨씬 코드가 깔끔하게 떨어지는것을 볼 수 있었다.

그래서 나도 코드를 수정했다.

import sys
n,m = map(int, sys.stdin.readline().split())
a = []
for _ in range(n):
    a.append(list(map(int,sys.stdin.readline().split())))
n2,m2 = map(int, sys.stdin.readline().split())
b=[]
for _ in range(n2):
    b.append(list(map(int,sys.stdin.readline().split())))
final = [[0 for i in range(m2)] for i in range(n)]
for i in range(n):
    for j in range(m2):
        for n in range(m):
            final[i][j] += a[i][n] * b[n][j]
            
for i in final:
    for j in i:
        print(j,end=' ')
    print()