[파이썬 PYTHON · 자바 JAVA] 프로그래머스 【구명보트】

---

자바 코드

import java.util.*;
class Solution {
    public int solution(int[] people, int limit) {
        int answer = 0;
        Arrays.sort(people);
        int left = 0;
        int right = people.length-1;
        
        while(left<=right){
            int sum = people[left]+people[right];
            if(left==right){
                answer++;
                break;
            }
            if(sum<=limit){
                answer++;
                left++;
                right--;
            }else{
                answer++;
                right--;
            }
        }
        return answer;
    }
}

파이썬 코드

def solution(people, limit):
    answer = 0
    people.sort()
    
    left = 0
    right = len(people)-1
    
    while left <= right:
        if left == right:
            answer+=1
            break
            
        _sum = people[left]+people[right]
        
        if _sum<=limit:
            answer+=1
            left+=1
            right-=1
        else:
            answer+=1
            right-=1
            
    return answer

LeetCode에서 포인터 방식 문제를 많이 풀어봐서 쉽게 해결했던 문제였다.

 

먼저 오름차순으로 정렬해준 뒤에, left = 0, right = 배열 끝 인덱스 로 잡아준다.

 

보트는 최대 2명이 탈 수 있다고 했으므로

가장 무거운 사람이 가장 가벼운 사람 순으로 1명씩 태울 수 있는 상황이면 태워가는게 보트가 가장 적게 움직일것이다.

 

만약 최대인원수가 안정해져있었으면 꽤나 까다로울뻔..

 

아무튼, 보트를 안 탄 사람 중 가장 무거운 사람과 가장 가벼운 사람이 무게 제한을 만족하면 같이 태워보내고

right 를 -1 감소 left를 +1 증가시킨다.

 

만약 제한을 넘는다면, 무거운 사람은 누구와도 함께 탈 수 없음이 확실해지기 때문에 먼저 태워보내야 하므로

right를 -1 감소 시킨다.

 

만약 left==right 되는 순간이 온다면, 1명만 남게 되는 것이므로 answer++ 시키면서 태워 보내면 정답이 나온다.