본문 바로가기

알고리즘/알고리즘 문제

[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

 

입출력 예 설명

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

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

 

 

<내가 푼 방법>

def solution(routes):
    # 결국에 교집합 갯수 구하는 문제
    # 앞자리 기준으로 오름차순으로 정렬한다.
    routes.sort(key=lambda x:x[0])
    # 무조건 1개는 들어갈 수 밖에 없으니 1개부터 시작한다.
    answer = 1 
    endpoint = routes[0][1]
    routes.pop(0)
    for route in routes:
        if route[0] <= endpoint:
            endpoint = min(route[1], endpoint)
        else:
            answer += 1
    return answer

이미 최초에 존재하는 answer의 갯수 1개는 새로운 교집합이 나타날 경우 +1 된다.

앞자리를 기준으로 정렬한 후,

첫번째 순서의 뒷자리 수가 다음 순서의 앞자리 보다 크면 교집합이 존재한다고 받아들인다.

그리고 첫번째 순서의 뒷자리 수와 두번째 순서의 뒷자리 수 중 더 작은것을 선택해 다시 뒷자리수의 기준점으로 삼는다. 이런 방식을 반복하며 기준점의 뒷자리 수보다 다음 순서의 첫번째 수가 클 때 현존하는 교집합을 벗어났다는 의미이므로 +1을 해준다.

 

실행결과에서는 통과되지만 효율성에서 탈락한다.

 

이유가 왜인지는 아직 잘 모르겠다.

 

def solution(routes):
    # 결국에 교집합 갯수 구하는 문제
    # 일단 앞자리 기준으로 오름차순으로 정렬한다.
    routes.sort(key=lambda x:x[0])
    answer = 1
    endpoint = routes[0][1]
    routes.pop(0)
    for route in routes:
        if route[0] <= endpoint:
            endpoint = min(route[1], endpoint)
        else:
            endpoint = route[1]
            answer += 1
    return answer

else에 기준값을 다시 초기화해주지 않아서 발생한 문제였다.

endpoind값을 주니 실행된다.

 

<다른사람이 푼 방법>

효율성에서 탈락하지 않는 코드이다.

앞자리를 기준으로 정렬하지 않고 뒷자리를 기준으로 정렬하여 풀었다.

확실히 훨씬 간단하다.