2740번: 행렬 곱셈
첫째 줄에 행렬 A의 크기 N 과 M이 주어진다. 둘째 줄부터 N개의 줄에 행렬 A의 원소 M개가 순서대로 주어진다. 그 다음 줄에는 행렬 B의 크기 M과 K가 주어진다. 이어서 M개의 줄에 행렬 B의 원소 K개
www.acmicpc.net
# #행렬 곱셈
N , M = map(int, input().split(' '))
a = []
b = []
for _ in range(N):
a.append(list(map(int, input().split(' '))))
M , K = map(int, input().split(' '))
for _ in range(M):
b.append(list(map(int, input().split(' '))))
result = [[0 for _ in range(K)] for _ in range(N)] #행렬 N*K 크기만큼
for n in range(N):
for k in range(K):
for m in range(M):
result[n][k] += a[n][m] * b[m][k] #!!
for i in result:
for j in i:
print(j, end = " ")
print()저 for 3중문이 너무 헷갈려서 고생했다...................ㅠㅠㅠㅠ 바본가 나는
그래도 다음에 풀어볼 때에는 수월하겠지!
반응형