(Python)백준 코딩테스트 연습 - 행렬 곱셈

1. 문제

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

2740번: 행렬 곱셈

2. 풀이

n, m = map(int, input().split())
a_list = []
for i in range(n):
    a_list.append(list(map(int, input().split())))
    
m, k = map(int, input().split())
b_list = [[] for i in range(k)]
for i in range(m):
    b = list(map(int, input().split()))
    for j in range(k):
        b_list[j].append(b[j])
# print('a_list :', a_list)
# print('b_list :', b_list)
result_list = [[] for i in range(n)]
for i, v in enumerate(a_list):
#     print(i, v)
    for j in b_list:
        p_sum = 0
        for p in range(len(v)):
            p_sum += v[p] * j[p]
        result_list[i].append(p_sum)
for result in result_list:
    for result_p in result:
        print(result_p, end=' ')
    print()