[프로그래머스] 구명보트 - Python

문제

---

코딩테스트 연습 - 구명보트

 

 

 

풀이

---

해당 문제는 그리디를 이용해 해결했습니다.

오름차순으로 people 배열을 정렬해준 뒤, 왼쪽 커서와 오른쪽 커서를 지정해줍니다.

인덱스에 해당하는 사람이 구명보트를 타고 떠나면, 인덱스를 옮겨줍니다.

 

왼쪽 커서와 오른쪽 커서가 같아질 때까지 반복문을 도는데, 

people[left] + people[right] 값이 limit 값을 넘으면 오른쪽 인덱스에 해당하는 사람만 보트에 보내고 (right -= 1), 

people[left] + people[right] 값이 limit 값과 같거나 넘지 않으면

왼쪽 인덱스, 오른쪽 인덱스에 해당하는 사람들 모두 보트에 태워 보냅니다 (left += 1, right -= 1). 

 

보트가 하나 떠날 때마다 카운트를 높여줍니다.

---

💡 풀이 코드는 아래와 같습니다. 

def solution(people, limit):
    people.sort()

    left = 0
    right = len(people) - 1
    count = 0

    while left <= right:
        if people[left] + people[right] > limit:
            right -= 1
            count += 1
        else:
            left += 1
            right -= 1
            count += 1

    return count

---

🔧 리팩토링한 코드는 아래와 같습니다. 

def solution(people, limit):
    people.sort()

    left = 0
    right = len(people) - 1
    count = 0

    while left <= right:
        if people[left] + people[right] <= limit:
            left += 1
        right -= 1
        count += 1

    return count