[프로그래머스/Python] 요격 시스템

문제 정보

제목 : 요격 시스템

난이도 : Lv.2

사용 언어 : Python

문제 링크

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

입출력 예

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

 

입출력 예 설명

위 그림과 같이 최소 세 번의 요격 미사일 발사로 전부 방어할 수 있습니다.

---

나의 풀이

B나라가 쏘는 미사일의 쏠 수 있는 범위를 나타내기위해 sB, eB를 선언하고 0으로 초기화합니다. targets를 정렬하여 반복하는 for루프를 이용합니다. 만약 B 미사일의 끝나는 범위가 A의 미사일 s구간보다 작거나 같으면 미사일 하나가 더 필요한 경우 이므로 answer에 1을 추가하고 sB는 sA로, eB는 eA로 새롭게 구간을 정합니다. 만약 해당 미사일로 해결이 가능한 경우엔 B의 s구간은 sB와 sA의 최대값으로, B의 e구간은 eB와 eA의 최소값으로 구간을 정합니다. 반복이 끝나면 answer을 return합니다.

 

코드

def solution(targets):
    answer = 0
    sB, eB = 0,0
    
    for sA,eA in sorted(targets) :
        if( eB <= sA ) : # 미사일 하나가 더 필요한 경우
            answer += 1
            sB = sA
            eB = eA
        else : # 해당 미사일로 해결 가능한 경우
            sB = max(sB, sA)
            eB = min(eB, eA)
            
    return answer