알고리즘/C++

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

지휘리릭 2020. 3. 7. 14:55
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<int> people, int limit) {
    int answer = 0;
    sort(people.begin(), people.end());
    int i = 0, j = people.size()-1;
    
    while(j > i){
        if(people[i] + people[j] <= limit) i++;
        answer += 1;
        j--;
        if(j == i){
            answer += 1;
            break;
        }
    }
    return answer;
}

i 는 0부터 시작하고 j는 제일 끝에서부터 시작해서 

가벼운 사람과 뚱뚱한 사람을 태우면 가장 최소 보트 개수를 얻을 수 있다.

만약 두 사람을 모두 태울 수 있다면, i += 1 을 하여 가벼운 사람을 옮겨준다. 

그렇지 않다면, 같이 탈 수 있는 j가 나타날 때 까지, j만 태워서 보낸다.

 

두 경우의 모두 answer += 1, j -= 1을 한다.

 

마지막에는, i j 가 딱 맞아 떨어지거나 i가 혼자 남는 경우 두 가지이다.

만약 i, j가 맞아 떨어지면 while문의 조건 i < j에 해당하지 않아서 while문을 나오게 되고

i가 혼자 남으면 j == i if문에 해당되고 혼자 남은 i를 태울 보트가 추가되고 while문을 나온다.