프로그래머스(탐욕법, Greedy)-단속카메라-Python

문제 설명

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

제한사항

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

입출력 예

routes	return
[[-20,15], [-14,-5], [-18,-13], [-5,-3]]	2

 

---

내 문제풀이

1. 고속도로에서 나간 지점이 가장 빠른 순으로 routes를 정렬한다.

2. 첫 번째 카메라는 정렬한 routes의 가장 첫 번째 차량이 나간 지점에 설치한다. 따라서 설치 카메라 개수(answer)은 1로 초기화를 한다.

3. 두 번째 차량부터 확인하며 반복문을 돈다. 만약 차량이 들어온 시점이 카메라 설치 지점보다 늦다면, 단속카메라와 만날 수 없다. 따라서 해당 경우에 새로운 카메라를 차량이 나가는 시점에 설치해준다.

내 코드 - 파이썬

def solution(routes):
    answer = 1
    routes.sort(key=lambda x:x[1])
    cam = routes[0][1]
    for idx in range(1, len(routes)):
        if (routes[idx][0] > cam):
            cam = routes[idx][1]
            answer += 1        
    
    return answer