[Beakjoon] 백준 1902번 Python
Beakjoon 1902, 배
solved.ac Gold5
Greedy 알고리즘
문제
지민이는 항구에서 일한다. 그리고 화물을 배에 실어야 한다. 모든 화물은 박스에 안에 넣어져 있다. 항구에는 크레인이 N대 있고, 1분에 박스를 하나씩 배에 실을 수 있다. 모든 크레인은 동시에 움직인다.
각 크레인은 무게 제한이 있다. 이 무게 제한보다 무거운 박스는 크레인으로 움직일 수 없다. 모든 박스를 배로 옮기는데 드는 시간의 최솟값을 구하는 프로그램을 작성하시오.
입력
첫째 줄에 N이 주어진다. N은 50보다 작거나 같은 자연수이다. 둘째 줄에는 각 크레인의 무게 제한이 주어진다. 이 값은 1,000,000보다 작거나 같다. 셋째 줄에는 박스의 수 M이 주어진다. M은 10,000보다 작거나 같은 자연수이다. 넷째 줄에는 각 박스의 무게가 주어진다. 이 값도 1,000,000보다 작거나 같은 자연수이다.
출력
첫째 줄에 모든 박스를 배로 옮기는데 드는 시간의 최솟값을 출력한다. 만약 모든 박스를 배로 옮길 수 없으면 -1을 출력한다.
Solution
n = int(input())
w = list(map(int, input().split()))
m = int(input())
b = list(map(int, input().split()))
w.sort(reverse=True)
b.sort(reverse=True)
count = 0
if w[0] < b[0]:
print("-1")
exit()
while len(b) != 0:
count += 1
i = 0
j = 0
while i < n and len(b) != 0:
while j < len(b):
if w[i] >= b[j]:
del b[j]
break
else:
j += 1
i += 1
print(count)
풀이
먼저 하중이 큰 크레인으로 무거운 박스를 잡아서 옮기는 것이 가장 최적이기 때문에 입력받은 크레인의 하중과 박스의 무게를 내림차순으로 정렬해줍니다. 크레인 중 가장 높은 하중보다 박스의 무게가 높을 경우 바로 -1을 출력합니다.
무거운 하중을 가진 크레인 순서대로 해당 크레인이 잡을수 있는 무게와 박스의 무게를 비교하여 가장 큰 무게의 박스를 배로 옮깁니다. 그리고 해당 박스는 list에서 제거합니다.
모든 크레인이 박스를 옮긴 여부와는 무관하게 한 사이클이 돌면 count를 1개 올려줍니다.
이 작업을 박스가 전부 없어질때까지 반복합니다.