Algorithm

[BOJ] 배 1092번 C++

따봉치치 2024. 4. 29. 14:53

문제

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

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

입력

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

출력

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

접근 방식

 

크레인과 박스 모두 vector로 넣어주고 오름차순 정렬한다

두 개의 최댓값을 비교해서 크레인이 들수 있는 최대 무게보다 박스가 무겁다면 -1을 출력하고 리턴한다

 

그리고 while문으로 박스가 비어있지 않을때까지 반복하면서

한 번에 모든 크레인이 들 수 있는 박스의 개수를 확인한다

그리고 옮긴 박스는  박스 vector에서 erase를 수행하고 cnt값을 증가해주면 된다!

 

코드

 

#include<bits/stdc++.h>
using namespace std;

int main() {
    ios::sync_with_stdio(false); cin.tie(0);

    int N, M;
    cin >> N;
    vector<int> k(N);
    for (int i = 0; i < N; i++) cin >> k[i];
    sort(k.rbegin(), k.rend());
    
    cin >> M;
    vector<int> b(M);
    for (int i = 0; i < M; i++) cin >> b[i];
    sort(b.rbegin(), b.rend()); 

    if (b[0] > k[0]) {
        cout << -1;
        return 0;
    }

    int cnt = 0;
    while (!b.empty()) {
        int idx = 0;
        for (int i = 0; i < N && !b.empty(); i++) {
            while (idx < b.size() && b[idx] > k[i]) {
                idx++;
            }
            if (idx < b.size()) {
                b.erase(b.begin() + idx);
            }
        }
        cnt++;
    }

    cout << cnt;
    return 0;
}