[Programmers] 단속 카메라 Java

문제 링크

https://school.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 이하입니다.

 

입출력 예
routes
[[-20,-15], [-14,-5], [-18,-13], [-5,-3]]

return

2

 

입출력 예 설명
-5 지점에 카메라를 설치하면 두 번째, 네 번째 차량이 카메라를 만납니다.
-15 지점에 카메라를 설치하면 첫 번째, 세 번째 차량이 카메라를 만납니다.

 

 

 

 

풀이

routes 입력 순서대로 그림을 그려봤다. 

 

차량이 나가기 직전까지만 단속 카메라를 마주치면 되기 때문에 나간 기준으로 정렬한다.

 

단위 구간 내 나가는 지점이 가장 빠른 차량을 기준으로 그 차량 범위 내에 포함되어 있는 차량들을 모두 단위 구간 내에 포함시킨다. 카메라는 가장 빨리 나가는 차량의 나가는 지점에 설치한다.

 

 

 

 

구현

int[][]를 나가는 지점을 기준으로 정렬하기 위해서 Arrays 클래스의 sort​(T[] a, Comparator<? super T> c) 메서드를 사용한다.

sort의 parameter는 비교하고 싶은 int 이차원 배열과 Comparator 인터페이스이다.

 

1차원 배열의 index 1번째를 비교하기 위해 compare​(T o1, T o2) 메서드를 Override하여 사용한다.

 

        Arrays.sort(routes, new Comparator<int[]>() {
        	@Override
        	public int compare(int[] route1, int[] route2) {
        		return route1[1] - route2[1];
        	}
        });

 

 

 

원래는 if(exit >= routes[index][0])만 체크했었는데 프로그래머스 반례를 보니 

 

 

이렇게 네 번째 차량의 경우의 수도 고려해야 했다. 따라서 && exit <= routes[index++][1] 이 조건을 추가하여 보완했다.

코드

package codingtest;

import java.util.Arrays;
import java.util.Comparator;

class Solution {
    public int solution(int[][] routes) {
        int answer = 1;
        
        Arrays.sort(routes, new Comparator<int[]>() {
            @Override
            public int compare(int[] route1, int[] route2) {
                return route1[1] - route2[1];
            }
        });
        
        int exit = routes[0][1];
        int index = 0; // 현재 차량의 인덱스
        int len = routes.length;

        while(index < len) {
        	// 차량의 끝점이 다음 차량 구간에 포함되는지 체크하기
        	if(exit >= routes[index][0] && exit <= routes[index++][1]) continue;
			else {
        		answer++;
        		exit = routes[index++][1];
        	}
        }
		return answer;
    }
    
    public static void main(String[] args) {
    	int[][] routes = {{-100, 100}, {50, 170}, {150, 200}, {-50, -10}, {10, 20}, {30, 40}};
    	Solution s = new Solution();
    	
    	System.out.println(s.solution(routes));
	}
}