단속카메라 (그리디) - PYTHON

문제

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

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

제한 사항

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

입출력 예

예 설명

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

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

풀이

처음 이 문제를 보니까 옛날에 백준에서 풀었던 회의실 배정 그 문제와 동일하다 생각해서 약간 수월하게 풀었다.

• 알고리즘
  1. 종료 지점을 기준으로 정렬
  2. 초기 위치를 가장 빠른 시작 지점보다 앞으로 설정 (초기화)
  3. routes 내의 요소를 반복하며 시작 지점보다 카메라가 작은지 확인.
  4. 작다면 카메라 추가하고 위치를 요소의 종료 지점으로 갱신해줌.

• 종료 지점으로 정렬하는 이유

백준 회의실 배정 문제를 풀었을 때, 시작 지점을 기준으로 정렬을 했었지만,, 이러한 반례가 존재하게 된다.

일찍 시작을 해도 종료시간이 늦어지면 위의 알고리즘상에서 cctv는 1개만 설치가 되는 에러를 범하게 되기 때문이다.

코드

def solution(routes):
    answer = 0
    cctv_pos = -30001
    routes.sort(key= lambda x: x[1])
    
    for route in routes:
        if cctv_pos < int(route[0]):
            answer += 1
            cctv_pos = route[1]
            
    return answer