[python] 탐욕법_단속카메라(3단계)

EunBi Na·2022년 4월 7일
0

문제 설명

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

고속도로를 이동하는 차량의 경로 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 지점에 카메라를 설치하면 첫 번째, 세 번째 차량이 카메라를 만납니다.

문제풀이

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

-> 차량 동선이 최대한 많이 겹치는 구간에 카메라를 설치하기!
1) routes를 차량이 나간 지점(route[1]) 기준으로 정렬합니다.
2) routes를 순회합니다.
3) 만약 prev_camera < route[0] 라면, 개수를 세고, prev_camera 위치를 route[1]로 합니다. 개수를 반환합니다.
이 때 중요한 것은 prev_camera를 -30000으로 먼저 초기화해줘야 한다는 점!

def solution(routes):
    # 1. route[1] 기준으로 정렬
    routes = sorted(routes, key=lambda route: route[1])
    # 2. -30000부터 카메라 위치를 찾습니다.
    prev_camera = -30000
    answer = 0

    # 3. routes를 순회합니다.
    for route in routes:
        # 1. 만약 최근 카메라가 route[0]보다 작은지 체크합니다.
        # 1-1. 작다면, 카메라를 한 개 더 세웁니다.
        # 1-2. 최근 카메라의 위치를 갱신합니다.
        if prev_camera < route[0]:
            answer += 1
            prev_camera = route[1]

    # 4. 카메라 개수를 반환합니다.
    return answer

링크텍스트

결론적으로 [-5, -3] 구간을 지나는 차량이 감시 카메라에 잡히지 않습니다. 따라서 카메라를 이 구간의 끝에 설치해줍니다.

routes = [[-18, -13], [-14, -5], [-5, -3], [-20, 15]]
prev_camera = -3 # 현재 설치된 카메라 위치
answer = 2 # 카메라 개수
profile
This is a velog that freely records the process I learn.

0개의 댓글