[프로그래머스] 구명보트 (C++)

우리누리·2023년 12월 4일
0

👓 문제 설명


무인도에 갇힌 사람들을 구명보트를 이용하여 구출하려고 합니다. 구명보트는 작아서 한 번에 최대 2명씩 밖에 탈 수 없고, 무게 제한도 있습니다.

예를 들어, 사람들의 몸무게가 [70kg, 50kg, 80kg, 50kg]이고 구명보트의 무게 제한이 100kg이라면 2번째 사람과 4번째 사람은 같이 탈 수 있지만 1번째 사람과 3번째 사람의 무게의 합은 150kg이므로 구명보트의 무게 제한을 초과하여 같이 탈 수 없습니다.

구명보트를 최대한 적게 사용하여 모든 사람을 구출하려고 합니다.

사람들의 몸무게를 담은 배열 people과 구명보트의 무게 제한 limit가 매개변수로 주어질 때, 모든 사람을 구출하기 위해 필요

입출력 예
people limit return
[70, 50, 80, 50] 100 3
[70, 80, 50] 100 3

💣 제한 사항

  • 무인도에 갇힌 사람은 1명 이상 50,000명 이하입니다.
  • 각 사람의 몸무게는 40kg 이상 240kg 이하입니다.
  • 구명보트의 무게 제한은 40kg 이상 240kg 이하입니다.
  • 구명보트의 무게 제한은 항상 사람들의 몸무게 중 최댓값보다 크게 주어지므로 사람들을 구출할 수 없는 경우는 없습니다.

🚨 접근 방법

우선 최대 2명밖에 탈 수 없는 제한이 걸려있으므로 가장 무거운 사람 + 가장 가벼운 사람의 조합의 가능 여부만 판단하면 된다.
만약 저 조건이 없었다면 가장 무거운 사람 + 가벼운 사람 중 가장 무거운 사람에 대해 탐색을 해야했을 것이다.
따라서 벡터를 내림차순으로 정렬 후 덱에 복사하여 맨 앞의 사람과 맨 뒤의 사람의 몸무게를 이용하여 문제를 해결했다.


🚈 코드

#include <string>
#include <vector>
#include <algorithm>
#include <queue>
#include <iostream>
using namespace std;

int solution(vector<int> people, int limit) {
    int answer = 0,sum=0;
    sort(people.begin(),people.end(),greater<int>());
    deque<int> dq(people.size());
    copy(people.begin(),people.end(),dq.begin());
    
    while(!dq.empty()){
        int heavy=dq.front();
        dq.pop_front();
        int light=dq.back();
        if(heavy+light<=limit){
            if(!dq.empty()){
                dq.pop_back();
            }
        }
        answer++;
    }

    return answer;
}

0개의 댓글