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

timo·2021년 1월 8일
0
post-thumbnail

문제 설명

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

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

제한사항

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

입출력 예

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

입출력 예 설명

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

나의 풀이

def solution(routes):
    routes.sort(key=lambda x:x[1])
    Q = {i for i in range(len(routes))} #아직 카메라가 찍히지 않은 루트
    answer = 0
    while len(Q) != 0:
        camera = min(Q)
        cameraPoint = routes[camera][1]
        removeList = []
        for route in Q:
            if cameraPoint >= routes[route][0] and cameraPoint <= routes[route][1]:
                removeList.append(route)
        for item in removeList:
            Q.remove(item)
        answer += 1
    return answer

Feedback

학교 '알고리즘'수업에서 배웠던 것이 도움됬다. 'Greedy'알고리즘 파트에서 회의실을 최대한 많이 사용하게 하는 알고리즘을 공부했다. 회의실 사용이 끝나는 순으로 정렬해서 해결하는 방식이었는데, 여기서 영감을 받았다. 회의실 배정은 최대한 겹치지 않아야 했고, 단속 카메라는 최대한 겹쳐야 한다는 점에서 비슷한 방법으로 풀면 되지 않을까 생각했다.

profile
Backend Developer

0개의 댓글