백준 1461. 도서관

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

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

ans = 0

plus_books = []
minus_books = []
for book in books:
    if book > 0:
        plus_books.append(book)
    else:
        minus_books.append(book)

# 양수와 음수 리스트에서 가장 큰 값 중 절대값이 가장 큰 값을 찾아 한 번만 이동
if plus_books and minus_books:
    max_distance = max(abs(plus_books[-1]), abs(minus_books[0]))
elif plus_books:
    max_distance = abs(plus_books[-1])
else:
    max_distance = abs(minus_books[0])

# 양수 처리
while plus_books:
    ans += plus_books[-1] * 2
    for _ in range(m):
        if not plus_books:
            break
        plus_books.pop()

# 음수 처리
while minus_books:
    ans += abs(minus_books[0]) * 2
    for _ in range(m):
        if not minus_books:
            break
        minus_books.pop(0)

# 왕복이 필요 없는 가장 먼 거리 한 번만 이동
ans -= max_distance

print(ans)

가장 먼 곳부터 왕복운동하는데 그동안 최대한 들 수 있는 m개의 책을 pop으로 제거시킴.

근데 가장 먼거리는 한번만 이동하므로 맨 처음 구해놨던 가장 먼거리를 한번 빼줌.