프로그래머스 - 단속카메라 / Python

문제: https://programmers.co.kr/learn/courses/30/lessons/42884

코딩테스트 연습 - 단속카메라

문제 설명

문제 설명

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

고속도로를 이동하는 차량의 경로 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):
    answer = 0
    cam = -30000
    routes.sort(key=lambda x: x[1])  # 진출지점 기준 정렬
    chk = [False] * len(routes)  # 카메라 단속 체크

    for idx, route in enumerate(routes):
        if not chk[idx]:  # 카메라를 만나지 않았으면 실행
            cam = route[1]  # 카메라를 해당 차량의 진출지점에 설치
            for i, r in enumerate(routes):
                if not chk[i] and r[0] <= cam <= r[1]:
                    chk[i] = True
            answer += 1
    return answer

---

풀이

def solution(routes):
    answer = 0
    cam = -30000
    routes.sort(key=lambda x: x[1])  # 진출지점 기준 정렬
    chk = [False] * len(routes)  # 카메라 단속 체크

    for idx, route in enumerate(routes):
        if not chk[idx]:  # 카메라를 만나지 않았으면 실행
            cam = route[1]  # 카메라를 해당 차량의 진출지점에 설치
            for i, r in enumerate(routes):
                if not chk[i] and r[0] <= cam <= r[1]:
                    chk[i] = True
            answer += 1
    return answer

 차량의 진입, 진출 지점들이 주어지면 모든 차량이 카메라를 한 번은 만나도록 설치하는 문제입니다.

이 문제에서의 핵심은 카메라를 최소로 설치하는 것입니다. 카메라를 최소로 설치하려면 진출 지점을 기준으로 정렬합니다. 그리고 정렬된 차량들의 진출 지점에 카메라를 놓는 것이 다른 곳에 놓는 것보다  다른 차량들과 겹칠 가능성이 높으므로 진출 지점마다 놓고, 만약 단속 카메라를 만난 적이 있으면 놓지 않는 식으로 구현합니다.

 

 제가 작성한 코드는 카메라를 각 차량의 단속 카메라 조우 여부를 저장한 리스트를 생성하고, 카메라를 놓을 때마다 겹치는 차량들을 확인하는 방식입니다. 

 

 

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

 이 코드는 '다른 사람의 풀이'를 보면 나오는 풀이입니다. 여기서는 카메라 조우 여부없이, 진출 지점에 카메라를 놓고 처음으로 내 진출 지점보다 진입지점이 먼 차량의 진출 지점에 카메라를 놓는 방식입니다. 문제 해결을 위한 아이디어는 비슷하나 코드를 구현함에 있어서 훨씬 간단하고 수행 시간도 훨씬 짧은 것을 볼 수 있습니다.