[백준] #2740 행렬 곱셈 python

2023. 4. 3. 14:39

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

 

2740번: 행렬 곱셈

첫째 줄에 행렬 A의 크기 N 과 M이 주어진다. 둘째 줄부터 N개의 줄에 행렬 A의 원소 M개가 순서대로 주어진다. 그 다음 줄에는 행렬 B의 크기 M과 K가 주어진다. 이어서 M개의 줄에 행렬 B의 원소 K개

www.acmicpc.net

 

📕 설명 📕

 

행렬 연산을 위해 리스트를 2개를 구성하고 3중 for문을 통해 연산을 진행하였다.

 

🧑🏻‍💻 나의 풀이 🧑🏻‍💻

 

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

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

for i in range(N):
    for j in range(K):
        for k in range(M):
            result[i][j] += A[i][k] * B[k][j]
for i in result:
    for j in i:
        print(j, end = ' ')
    print()

BELATED ARTICLES

more