[Programmers] 요격 시스템 ( PriorityQueue )

 

- 목차

 

문제 설명.

 

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 함수를 완성해 주세요.

---

제한 사항

• 1 ≤ targets의 길이 ≤ 500,000
• targets의 각 행은 [s,e] 형태입니다.
  • 이는 한 폭격 미사일의 x 좌표 범위를 나타내며, 개구간 (s, e)에서 요격해야 합니다.
  • 0 ≤ s < e ≤ 100,000,000

 

문제 분석.

이 문제는 최소한의 요격 미사일로 모든 타겟을 요격해야하는 문제입니다.

그래서 먼저 모든 타겟들을 종료 시점을 기준으로 오름차순 정렬합니다.

왜냐하면 종료 시점이 빠를수록 가장 먼저 요격을 해야하기 때문입니다.

그래서 가장 빨리 종료되는 타겟을 기준으로 이 타겟과 범위를 공유하는 다른 타겟들을 찾아야합니다.

 

다른 예시로 섭취 가능 기간이 정해진 여러 오래된 음식들이 존재할 때,

음식이 상하기 이전에 최소한의 끼니로 모든 오래된 음식을 섭취하려고 한다고 예시를 들어보겠습니다.

이 경우에도 가장 빨리 상할 것 같은 음식을 기준으로 섭취 가능 기간을 공유하는 다른 음식들을 같이 먹는게 효율적입니다.

 

그래서 저는 종료 시점을 기준으로하는 PriorityQueue 를 사용하였습니다.

 

문제 풀이.

import java.util.*;

class Solution {
  public int solution(int[][] targets) {
    int answer = 0;
    PriorityQueue<int[]> missileComingSchedule = new PriorityQueue<>((o1, o2) -> o1[1] - o2[1]);

    for (int i = 0; i < targets.length; i++) {
      missileComingSchedule.add(targets[i]);
    }

    int[] baseTarget = null;
    while (true) {
      if (missileComingSchedule.isEmpty()) break;
      if (baseTarget == null) {
        baseTarget = missileComingSchedule.poll();
        answer++;
        continue;
      }

      int[] missile = missileComingSchedule.peek();

      // 동시에 요격이 가능하다면
      if (baseTarget[1] > missile[0]) {
        // 두번째 미사일도 함께 제거.
        missileComingSchedule.poll();
      } else {
        baseTarget = null;
      }
    }

    return answer;
  }
}