본문 바로가기
python-algorithm

[백준] 2740

by 무적김두칠 2021. 2. 23.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
x=[]
y=[]
xa,xb=map(int, input().split())
for i in range(xa):
    x.append(list(map(int, input().split())))
ya,yb=map(int, input().split())
for i in range(ya):
    y.append(list(map(int, input().split())))
result=[]
for i in range(xa):
    result.append([0]*yb)
for i in range(len(x)):
   for j in range(len(y[0])):
       for k in range(len(y)):
           result[i][j] += x[i][k] * y[k][j]
for r in result:
   print(*r)
cs

행렬 곱연산은 쉽게 구현했는데..
행렬 초기화를 잘 못해서 계속 에러가 났네요

numpy의 dot 이용하면 더 쉽게 구할 수 있습니다~

반응형

'python-algorithm' 카테고리의 다른 글

[백준] 2997  (0) 2021.02.26
[백준] 2765  (0) 2021.02.25
[백준] 2530  (0) 2021.02.23
[백준] 2163  (0) 2021.02.23
[백준] 2161  (0) 2021.02.23

댓글