[Programmers/Python] 탐욕법 - 단속카메라

Programmers - 단속카메라

문제

고속도로를 이동하는 모든 차량이 고속도로를 이용하면서 단속용 카메라를 한 번은 만나도록 카메라를 설치하려고 합니다.

고속도로를 이동하는 차량의 경로 routes가 매개변수로 주어질 때, 모든 차량이 한 번은 단속용 카메라를 만나도록 하려면 최소 몇 대의 카메라를 설치해야 하는지를 return 하도록 solution 함수를 완성하세요.

제한사항

• 차량의 대수는 1대 이상 10,000대 이하입니다.
• routes에는 차량의 이동 경로가 포함되어 있으며 routes[i][0]에는 i번째 차량이 고속도로에 진입한 지점, routes[i][1]에는 i번째 차량이 고속도로에서 나간 지점이 적혀 있습니다.
• 차량의 진입/진출 지점에 카메라가 설치되어 있어도 카메라를 만난것으로 간주합니다.
• 차량의 진입 지점, 진출 지점은 -30,000 이상 30,000 이하입니다.

입출력 예

입출력 예 설명

• 5 지점에 카메라를 설치하면 두 번째, 네 번째 차량이 카메라를 만납니다.
• 15 지점에 카메라를 설치하면 첫 번째, 세 번째 차량이 카메라를 만납니다.

풀이

설계

1. routes의 요소들을 '진출 지점(routes[1])'을 기준으로 오름차순 정렬한다.
2. 카메라의 최초 위치를 -30001로 초기화한다.
3. routes를 순회하면서 현재 카메라의 위치가 진입 지점(routes[0])보다 작은지 비교한다.
   • 작은 경우 차량을 단속하지 못한 것이므로, 카메라를 추가로 세운다. 이때 추가하는 카메라의 위치는 진출 지점으로 한다(앞서 routes를 진출 지점 순으로 정렬했으므로 이후의 차량들은 이 카메라보다 앞서 진출하는 경우는 없다).
   • 큰 경우 차량을 단속한 것이므로 패스한다.

코드

def solution(routes):
    routes_sorted = sorted(routes, key=lambda x: x[1])
    num_cameras = 0
    current_camera_location = -30001

    for route in routes_sorted:
        if route[0] > current_camera_location:
            num_cameras += 1
            current_camera_location = route[1]

    return num_cameras