Softeer: 복잡한 조립라인2

2021. 10. 19. 18:04Algorithm

    목차
반응형
import collections
import sys


K, N = map(int, sys.stdin.readline().split())
grid = []
tran = []

for i in range(N - 1):
    line = list(map(int, sys.stdin.readline().split()))
    grid += line[:K],
    tran += line[K],

line = list(map(int, sys.stdin.readline().split()))
grid += line,

for i in range(N - 1):
    mn = min(grid[i])
    idx = grid[i].index(mn)

    for j in range(K):
        if j == idx:
            grid[i + 1][j] += mn
        else:
            grid[i + 1][j] += min(grid[i][j], mn + tran[i])

print(min(grid[-1]))
반응형

'Algorithm' 카테고리의 다른 글

알고리즘 정의 (Algorithm)  (0) 2021.12.23
Softeer: 수퍼바이러스  (0) 2021.10.21
Softeer: 복잡한 조립라인1  (0) 2021.10.19
Softeer: 로봇이 지나간 경로  (0) 2021.10.15
Softeer: 징검다리2  (0) 2021.10.14