[프로그래머스/파이썬] (탐욕법(Greedy)) 단속카메라

코라닝·2021년 5월 4일
1

프로그래머스

목록 보기
23/35

출처

문제 설명

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

고속도로를 이동하는 차량의 경로 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_sorted = sorted(routes)
    cnt = 1
    start = routes_sorted[0][0]
    end = routes_sorted[0][1]
    for i, route in enumerate(routes_sorted):
        if route[0] <= end:
            start = route[0]
            end = min(end, route[1])
        else:
            cnt += 1
            start = route[0]
            end = route[1]
    return cnt

효율성 테스트: ~2.47ms / 10.7MB

앞서 푼 [섬 연결하기]에서 고전을 해서 그런지 이번 문제는 평이하단 느낌이었다.

기본적인 풀이는 routes를 진입 시점을 기준으로 정렬하여 차량들이 최대한 같은 카메라를 만날 수 있을 때까지 순서대로 탐색한 뒤, 어느 순서에서 새로운 카메라가 필요하다면 cnt를 1 더하고 카메라를 설치할 수 있는 지점인 start와 end를 그 순서의 진입 시점과 나간 시점으로 설정한다.

start와 end는 차량들의 진입 시점과 나간 시점을 고려하여 현재의 카메라를 설치할 수 있는 지점의 시작점과 끝점을 의미한다.
만약에 다음 순서의 차량의 진입시점이 end보다 작다면 이 차량 또한 같은 카메라를 만나고, 이에 따라 이 차량의 진입 시점을 start에 입력하고 만약에 나간 시점이 end보다 빠르다면 end를 이 차량의 나간 시점으로 두면 된다.

만약에 진입 시점이 end보다 크다면, 새로운 카메라가 필요하므로 앞서 말한 듯이 cnt에 1을 더하고, 새 카메라의 시작, 끝 점은 각각 현재 순서의 차량의 진입 시점과 나간 시점으로 초기화한다.

다른 풀이

def solution(routes):
    routes = sorted(routes, key=lambda x: x[1])
    last_camera = -30000

    answer = 0

    for route in routes:
        if last_camera < route[0]:
            answer += 1
            last_camera = route[1]

    return answer

효율성 테스트: ~1.00ms / 10.7MB

이 풀이는 우선 routes를 나가는 지점을 기준으로 정렬한다.
그리고 새로운 카메라가 필요하게 되면 그 때의 차량이 나가는 지점에 카메라를 설치한다고 가정한다.
그리고 그 뒤에 오는 차량들이 이 카메라를 만나지 않는다면, 즉 카메라의 위치가 차량의 진입 지점보다 앞에 있다면 새로운 카메라를 설치하고 이 때의 차량의 나가는 지점을 새로운 카메라의 위치로 설정한다.

처음에 last_camera를 -30000으로 두는 것은 -∞와 같다.
차량의 진입, 진출 시점을 -30000이상 30000이하로 뒀기 때문이다.

0개의 댓글