[Python/BOJ] 1092. 배

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

문제

지민이는 항구에서 일한다. 그리고 화물을 배에 실어야 한다. 모든 화물은 박스에 안에 넣어져 있다. 항구에는 크레인이 N대 있고, 1분에 박스를 하나씩 배에 실을 수 있다. 모든 크레인은 동시에 움직인다.

각 크레인은 무게 제한이 있다. 이 무게 제한보다 무거운 박스는 크레인으로 움직일 수 없다. 모든 박스를 배로 옮기는데 드는 시간의 최솟값을 구하는 프로그램을 작성하시오.

입력

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

출력

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

풀이

Code(PyPy)

import sys
input = sys.stdin.readline

n = int(input())
crane = list(map(int, input().split()))
m = int(input())
weights = list(map(int, input().split()))

# 1. 짐을 나를 수 없는 경우
if max(crane) < max(weights) : print(-1)
# 2. 이외의 경우
else :
    # 2-1. 크레인 역정렬
    crane.sort(reverse = True)
    # 2-2. 짐 역정렬
    weights.sort(reverse = True)
    # 2-3. 출력 변수 생성
    ans = 0
    # 2-4.
    while weights :
        # 2-4-1. 출력 변수 업데이트
        ans += 1
        # 2-4-2.
        for crane_idx, c in enumerate(crane) :
            for weight_idx, weight in enumerate(weights) :
                # 만약 크레인이 해당 짐을 들 수 있을 경우
                if c >= weight :
                    # 짐 제거
                    weights.pop(weight_idx)
                    break
    # 2-5. 결과 출력
    print(ans)