99클럽 코테 스터디 11일차 TIL - 도서관(백준 #1461)

❇️오늘의 학습 키워드 : 도서관(백준 #1461)

문제

세준이는 도서관에서 일한다. 도서관의 개방시간이 끝나서 세준이는 사람들이 마구 놓은 책을 다시 가져다 놓아야 한다. 세준이는 현재 0에 있고, 사람들이 마구 놓은 책도 전부 0에 있다. 각 책들의 원래 위치가 주어질 때, 책을 모두 제자리에 놔둘 때 드는 최소 걸음 수를 계산하는 프로그램을 작성하시오. 세준이는 한 걸음에 좌표 1칸씩 가며, 책의 원래 위치는 정수 좌표이다. 책을 모두 제자리에 놔둔 후에는 다시 0으로 돌아올 필요는 없다. 그리고 세준이는 한 번에 최대 M권의 책을 들 수 있다.

입력

첫째 줄에 책의 개수 N과, 세준이가 한 번에 들 수 있는 책의 개수 M이 주어진다. 둘째 줄에는 책의 위치가 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 책의 위치는 0이 아니며, 절댓값은 10,000보다 작거나 같은 정수이다.

출력

첫째 줄에 정답을 출력한다.

예제 입력 1 

7 2
-37 2 -6 -39 -29 11 -28

예제 출력 1 

131

예제 입력 2 

8 3
-18 -9 -4 50 22 -26 40 -45

예제 출력 2 

158

예제 입력 3 

6 2
3 4 5 6 11 -1

예제 출력 3 

29

예제 입력 4 

1 50
1

예제 출력 4 

1

정답 코드

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)

❇️오늘의 회고

  - 가장 먼 친구는 한번만 계산해야된다는 것을 한번에 떠올리지 못했다...ㅠ

  - 꾸준한 연습만이 답이다! 포기하지 말자!

  - 내일 11시, 문제를 풀기 전에 위 개념을 한번 더 복습하고 내일 문제에 사용된 알고리즘에 대해서 공부할 것이다!🙂