알고리즘 공부/탐욕알고리즘(Greedy)

[프로그래머스] 단속카메라

kdhoooon 2021. 6. 7. 15:52

문제


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

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

 

 

[제한사항]

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

 

 

 

 

풀이


우선 주어진 예제를 가지고 문제를 푸는 순서를  설명하겠다.

 

우선 끝나는 시간을 기준으로 정렬을 한다.

위의 그림처럼 정렬이 된다.

 

이렇게 정렬된 배열을 끝나는 시간을 기준으로 카메라를 설치한다.

만약 다음 구간의 start 가 lastCamera 보다 작다면 해당 구간은 카메라를 한번 만나는 것이므로 넘어간다.

다음 구간의 start 가 lastCamera보다 크면 해당 구간은 카메라를 한번도 만나지 않은 것이기 때문에 이 구간의 end 로 lastCamera 값을 초기화 시키고 answer++ 을 해줘 카메라 수를 늘려준다.

위 그림을 보면 이해하기 편하다.

 

<전체코드>

import java.util.*;

class Solution {
    
    static class car implements Comparable<car>{
        int start;
        int end;
        
        car(int start, int end){
            this.start = start;
            this.end = end;
        }
        
        @Override
        public int compareTo(car c){
            return this.end - c.end;
        }
    }
    public int solution(int[][] routes) {
        int answer = 0;        
            
        List<car> carList = new ArrayList<car>();
        for(int i = 0 ; i < routes.length ; i++){
            carList.add(new car(routes[i][0], routes[i][1]));
        }
        
        Collections.sort(carList);
        
        int lastCamera = carList.get(0).end;
        for(int i = 0 ; i< routes.length ; i++){
            if(i == 0){
                lastCamera = carList.get(i).end;
                answer++;
            }
            else if(lastCamera < carList.get(i).start){
                lastCamera = carList.get(i).end;
                answer++;
            }
            
        }
        
        return answer;
    }
}