[백준] 1461. 도서관 | 파이썬

백준 1461. 도서관 - 골드IV

문제 설명

일직선 좌표가 있고, 0 위치에 어질러진 책들을 각각 원래 자리에 가져다 둬야 하는데 최소한으로 움직여야 하고, 최소한의 거리를 출력하는 문제

- 현재 위치: 0
- 한번에 m권

---

---

파이썬 코드

n, m = map(int, input().split())
books = list(map(int, input().split()))

pos_li = []
neg_li = []
furthest = 0
result = 0

for book in books:
    furthest = max(abs(book), furthest)
    if book > 0:
        pos_li.append(book)
    else:
        neg_li.append(abs(book))

pos_li.sort(reverse=1)
neg_li.sort(reverse=1)

for i in range(0, len(pos_li), m):
    result += pos_li[i] * 2

for i in range(0, len(neg_li), m):
    result += neg_li[i] * 2

print(result - furthest)