[백준] 2740번 행렬곱셈

출처: [백준] 2740번 행렬곱셈

---

시간 제한	메모리 제한	제출	정답	맞은 사람	정답 비율
1 초	128 MB	6980	4853	4230	71.610%

---

문제

N x M크기의 행렬 A와 M x K크기의 행렬 B가 주어졌을 때, 두 행렬을 곱하는 프로그램을 작성하시오.

---

입력

첫째 줄에 행렬 A의 크기 N 과 M이 주어진다. 둘째 줄부터 N개의 줄에 행렬 A의 원소 M개가 순서대로 주어진다. 그 다음 줄에는 행렬 B의 크기 M과 K가 주어진다. 이어서 M개의 줄에 행렬 B의 원소 K개가 차례대로 주어진다. N과 M, 그리고 K는 100보다 작거나 같고, 행렬의 원소는 절댓값이 100보다 작거나 같은 정수이다.

---

출력

첫째 줄부터 N개의 줄에 행렬 A와 B를 곱한 행렬을 출력한다. 행렬의 각 원소는 공백으로 구분한다.

---

예제 입력 1

3 2
1 2
3 4
5 6
2 3
-1 -2 0
0 0 3

예제 출력 1

-1 -2 6
-3 -6 12
-5 -10 18

---

힌트

• -

---

출처

• 잘못된 조건을 찾은 사람: bupjae
• 문제의 오타를 찾은 사람: ekdns7952
• 잘못된 데이터를 찾은 사람: WeissBlume

---

알고리즘 분류

• 수학
• 구현

---

시간 제한

• -

---

풀이

• -

---

소스코드

import sys

input = sys.stdin.readline

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

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

result_matrix = [[0 for _ in range(K)] for _ in range(N)]

for n in range(N):
    for k in range(K):
        for m in range(M):
            result_matrix[n][k] += A_matrix[n][m] * B_matrix[m][k]

for row in result_matrix:
    for value in row:
        print(value, end=" ")
    print()