본문 바로가기

코딩 테스트/Lv.2

[프로그래머스] 요격 시스템 with Kotlin

문제

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

 

입출력 예

targets result
[[4,5],[4,8],[10,14],[11,13],[5,12],[3,7],[1,4]] 3

 

 

풀이 과정

1. targets[0]을 기준으로 오름차순 정렬

2. targets을 돌면서 이전 범위와 중첩되는 부분 탐색 후 저장

3. 만일 중첩되는 부분이 없다면 answer+1 이후 해당 미사일부터 다시 시작

  1. 첫 미사일: 1부터 4, 이전 범위는 없음으로 범위는 1 to 4
  2. 두번째 미사일 : 3 부터 7, 이전 범위와 중첩되는 부분은 3 to 4
  3. 세번째 미사일 : 4 부터 5, 이전 범위와 중첩되지 않음으로 answer++ 및 범위는 4 to 5로 시작
  4. 네번째 미사일 : 4 부터 8, 이전 범위와 중첩되는 부분은 4 to 5
  5. 위와 동일

 

코드

class Solution {
    fun solution(targets: Array<IntArray>): Int {
        var answer: Int = 1
        
        // Index 0 기준으로 sort
        targets.sortBy{it[0]}
        
        var range = 0 to 0 // 중첩 범위 저장
        for(target in targets){
            // 첫 타겟
            if(range == 0 to 0) {
                range = target[0] to target[1]
                continue
            }
            
            // 중첩
            if(target[0] < range.second){
                range = Math.max(range.first, target[0]) to Math.min(range.second, target[1])
            }
            else{
                answer++
                range = target[0] to target[1]
            }
        }
        
        return answer
    }
}