[프로그래머스] 요격 시스템 - Java

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/181188

문제 설명

A 나라가 B 나라를 침공하였습니다. B 나라의 대부분의 전략 자원은 아이기스 군사 기지에 집중되어 있기 때문에 A 나라는 B 나라의 아이기스 군사 기지에 융단폭격을 가했습니다.
A 나라의 공격에 대항하여 아이기스 군사 기지에서는 무수히 쏟아지는 폭격 미사일들을 요격하려고 합니다. 이곳에는 백발백중을 자랑하는 요격 시스템이 있지만 운용 비용이 상당하기 때문에 미사일을 최소로 사용해서 모든 폭격 미사일을 요격하려 합니다.
A 나라와 B 나라가 싸우고 있는 이 세계는 2 차원 공간으로 이루어져 있습니다. A 나라가 발사한 폭격 미사일은 x 축에 평행한 직선 형태의 모양이며 개구간을 나타내는 정수 쌍 (s, e) 형태로 표현됩니다. B 나라는 특정 x 좌표에서 y 축에 수평이 되도록 미사일을 발사하며, 발사된 미사일은 해당 x 좌표에 걸쳐있는 모든 폭격 미사일을 관통하여 한 번에 요격할 수 있습니다. 단, 개구간 (s, e)로 표현되는 폭격 미사일은 s와 e에서 발사하는 요격 미사일로는 요격할 수 없습니다. 요격 미사일은 실수인 x 좌표에서도 발사할 수 있습니다.
각 폭격 미사일의 x 좌표 범위 목록 targets이 매개변수로 주어질 때, 모든 폭격 미사일을 요격하기 위해 필요한 요격 미사일 수의 최솟값을 return 하도록 solution 함수를 완성해 주세요.

제한 사항

입출력 예

알고리즘: 그리디, 조건분기

풀이

최소로 사용해서 모든 폭격 미사일을 요격해야 한다. 그렇게 하려면 최대한 폭격 미사일들의 범위를 겹쳐서 요격해야 한다.

이런 식으로

그래서 결론은 겹치는 범위를 구해서 range를 적절히 업데이트해 주는 게 이 문제의 핵심이다.

먼저 targets 2차원 배열을 s를 기준으로 오름차순 정렬한다.
range 참조 변수를 선언해 준다. (이 변수는 겹치는 범위를 저장하기 위한 변수)
targets 반복문을 돌면서 range.s <= targets[i][0] < range.e 이 범위안에 들어오는지 확인한다.

true라면 같이 요격할 수 있는 범위이다. range를 업데이트해 줘야 하는데 s는 무조건 targets[i][0]이다. e는 range.e와 targets[i][1]중 작은 값으로 업데이트해 준다.

false라면 같이 요격할 수 없는 범위이다. range를 업데이트해 줘야 하는데 이전 범위와 관련 없는 범위이기 때문에 요격 미사일을 사용해야 한다. answer += 1 해 주고, 새로운 범위인 range = new Range(targets[i][0], targets[i][1])로 업데이트해 준다.

answer값에는 마지막 인덱스 범위에 미사일이 포함되어 있지 않기 때문에 +1 해 준다.

소스 코드

import java.util.*;
class Range {
    int s,e;
    Range(int s, int e) {
        this.s = s;
        this.e = e;
    }
}

class Solution {
    public int solution(int[][] targets) {
        int answer = 0;
        Arrays.sort(targets, (t1, t2) -> {
            return t1[0] - t2[0];
        });
        Range range = new Range(0, targets[targets.length-1][1]); //최대 범위
        for(int i=0; i<targets.length; i++) {
            if((range.s <= targets[i][0]) && (targets[i][0] < range.e)) {
                range = new Range(targets[i][0], Math.min(range.e, targets[i][1]));
            } else {
                answer += 1;
                range = new Range(targets[i][0], targets[i][1]);
            }
        }
        return answer + 1;
    }
}