본문 바로가기
개발/문제풀이

프로그래머스 '요격 시스템' 파이썬 풀이

by beomcoder 2023. 5. 17.
728x90
반응형
https://school.programmers.co.kr/learn/courses/30/lessons/181188
 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

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

def solution(t):
    lines, overlap = 0, {'s': 0, 'e': 0}
    
    for s, e in sorted(t):
        if overlap['e'] <= s:
            lines += 1
            overlap = {'s': s, 'e': e}
        else:
            overlap = {'s': max(overlap['s'], s), 'e': min(overlap['e'], e)}
            
    return lines

"""
처음 문제를 보고 푸는 방식을 상대 미사일 범위가 
겹치는 부분이 없을때마다 미사일을 하나씩 추가해야겠다고 생각했다.
그래서 미사일범위리스트를 정렬하여 하나씩 for문을 돌렸다.

겹치는 부분이 없다면 (if overlap['e'] <= s:) 미사일을 하나 추가하고,
새로운 미사일에 겹치는 부분을 기록하였다.

겹치는 부분이 있다면 현재 겹쳐지는 부분과 새로 요격해야할 미사일의 범위를
비교하여 겹쳐지는 부위를 새롭게 적어야 한다.

그래서 겹쳐지는 시작점은 둘 중에 더 큰 시작점이 되어야 하고,
겹쳐지는 끝점은 둘 중에 더 작은 끝점이 되어야 한다.

-
if overlap['e'] <= s 에 대해 설명하면
타겟리스트를 정렬하게 된다면 시작점이 작은순부터 정렬이 된다.
그래서 범위의 끝보다 요격해야할 미사일의 시작점이 크다면 겹치는게 아니다.
-

혹시 내 글이 이해가 안될 수도 있기 때문에 예시를 들어본다.
입출력 예인 targets = [[4,5],[4,8],[10,14],[11,13],[5,12],[3,7],[1,4]] 라면

먼저 정렬하면 [[1, 4], [3, 7], [4, 5], [4, 8], [5, 12], [10, 14], [11, 13]]
1, 4를 꺼내와서 overlap과 비교한다.

처음에는 overlap 된게 없으니(0,0) 미사일을 하나 추가하고
1, 4가 겹쳐지는 미사일 범위가 된다.
현재는 하나밖에 없으니까 선 하나가 전부 겹쳐진다고 생각하면 된다.

3, 7을 꺼내와서 overlap과 비교한다.
현재 1~4까지가 겹쳐지는 범위였고, 이는 3~7과 겹쳐지는 부분이 있다.
그럼 1~4 와 3~7의 겹쳐지는 부분은 3~4가 된다.
이걸 {'s': max(overlap['s'], s), 'e': min(overlap['e'], e)}라고 적었다.

4, 5를 꺼내와서 overlap과 비교한다.
양 끝점은 겹쳐질수 없다고 하였으니 3~4와 4~5는 겹치는게 없다.
그래서 새로운 미사일을 추가한다. 2번째 미사일에 겹쳐지는 부분은 4~5이다.

4, 8을 꺼내와서 overlap과 비교한다.
4~5와 4~8은 겹치는 부분이 있다. 겹치는 부분은 그대로 4~5가 된다.

5, 12는 4~5와 겹치지 않기 때문에 새로운 미사일을 추가한다.
그러면 3번째 미사일과 겹쳐지는 부분은 5~12가 되게 된다.

10, 14는 5~12와 겹쳐진다.
겹쳐지는 범위는 10~12가 되게 된다.

11, 13은 10~12와 겹쳐지는 범위가 있으므로 겹쳐지는 범위는 11~12이다.

for문을 다 돌았으니 추가한 미사일들을 리턴해준다.
"""

 

728x90
반응형

댓글