코딩테스트

[파이썬] 백준 1092번: 배 (그리디 알고리즘)

갓 시작한 코린이 2022. 12. 25. 18:23

 

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

 

1092번: 배

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

www.acmicpc.net

 

문제

지민이는 항구에서 일한다. 그리고 화물을 배에 실어야 한다. 모든 화물은 박스에 안에 넣어져 있다. 항구에는 크레인이 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

예제 입력 2 복사

2
19 20
7
14 12 16 19 16 1 5

예제 출력 2 복사

4

 

# 그리디 알고리즘
import sys

n= int(input())
cranes=list(map(int,input().split()))

m=int(input())
boxes=list(map(int,input().split()))

# 모든 박스를 옮길 수 없는 경우
if max(cranes) < max(boxes):
    print(-1)
    sys.exit()
    
# 각 크레인이 현재 옮겨야 하는 박스의 번호(0부터 시작)
positions=[0]*n
# 각 박스를 옮겼는지의 여부
check=[False]*m
# 최적의 해를 구하기위해 내림차순 정렬
cranes.sort(reverse=True)
boxes.sort(reverse=True)

result=0
count=0
while True:
    if count==len(boxes): # 박스를 다옮겼으면 종료
        break
    for i in range(n): # 모든 크레인에 대하여 각각 처리
        while positions[i]<len(boxes):
            # 아직 안 옮긴 박스 중에서 옮길수 있는 박스를 만날때까지
            if not check[positions[i]] and cranes[i]>=boxes[positions[i]]:
                check[positions[i]]=True
                positions[i]+=1
                count+=1
                break
            positions[i]+=1
    result+=1
print(result)