백준 온라인 저지, 그리디 / 1092번: 배 (파이썬 / 백준 골드문제)

문제

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

문제 정의

지민이는 항구에서 일한다. 그리고 화물을 배에 실어야 한다. 모든 화물은 박스에 안에 넣어져 있다. 항구에는 크레인이 N대 있고, 1분에 박스를 하나씩 배에 실을 수 있다. 모든 크레인은 동시에 움직인다.
각 크레인은 무게 제한이 있다. 이 무게 제한보다 무거운 박스는 크레인으로 움직일 수 없다. 모든 박스를 배로 옮기는데 드는 시간의 최솟값을 구하는 프로그램을 작성하시오.  

입력

첫째 줄에 N이 주어진다. N은 50보다 작거나 같은 자연수이다. 둘째 줄에는 각 크레인의 무게 제한이 주어진다. 이 값은 1,000,000보다 작거나 같다. 셋째 줄에는 박스의 수 M이 주어진다. M은 10,000보다 작거나 같은 자연수이다. 넷째 줄에는 각 박스의 무게가 주어진다. 이 값도 1,000,000보다 작거나 같은 자연수이다.

출력

첫째 줄에 모든 박스를 배로 옮기는데 드는 시간의 최솟값을 출력한다. 만약 모든 박스를 배로 옮길 수 없으면 -1을 출력한다.

예제 입력 1

3
6 8 9
5
2 5 2 4 7

예제 출력 1

2

접근 방법

- 0. 주어진 박스의 크기와 크레인의 무게를 내림차순으로 정렬한다.
- 1. min heap을 통해 각 크레인 당 박스를 옮긴 횟수를 저장하여 옮긴 박스의 갯수를 기준으로 만든다.
- 2. 크레인의 무게를 기준으로 한 리스트의 인덱스를 기준으로 현재 박스를 옮길 수 있으면 해당 크레인의 정보를 min heap에 삽입한다.
- 3. 박스를 옮긴 횟수를 매번 갱신해주며 모든 탐색이 끝났을 때 박스를 옮긴 최대값을 출력한다.

코드

# 접근 방법
# 0. 주어진 박스의 크기와 크레인의 무게를 내림차순으로 정렬한다.
# 1. min heap을 통해 각 크레인 당 박스를 옮긴 횟수를 저장하여 옮긴 박스의 갯수를 기준으로 만든다.
# 2. 크레인의 무게를 기준으로 한 리스트의 인덱스를 기준으로 현재 박스를 옮길 수 있으면 해당 크레인의 정보를 min heap에 삽입한다.
# 3. 박스를 옮긴 횟수를 매번 갱신해주며 모든 탐색이 끝났을 때 박스를 옮긴 최대값을 출력한다.
import heapq
n = int(input())
crane = list(map(int, input().split()))
m = int(input())
box = list(map(int, input().split()))
crane.sort(reverse=True)
box.sort(reverse=True)

i = 0
min_heap = []
result = 0
for x in box:
    while n > i and crane[i] >= x:
        heapq.heappush(min_heap, [0, crane[i]])
        i += 1
    if not min_heap:
        result = -1
        break
    count, c = heapq.heappop(min_heap)
    count += 1
    heapq.heappush(min_heap, [count, c])


while min_heap:
    x = heapq.heappop(min_heap)
    result = max(result, x[0])
print(result)