주니어 데이터 엔지니어 우솨's 개발일지

코딩 테스트(Python) 요격 시스템 본문

코딩 테스트

코딩 테스트(Python) 요격 시스템

우솨 2024. 12. 17. 01:52

문제 설명

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(targets):
    # 요격 미사일의 개수를 저장하는 변수
    shot = 0
    # 요격한 타겟의 개수를 저장하는 변수
    count = 0

    # 타겟을 끝 지점 기준으로 오름차순 정렬
    # x[1]은 각 타겟의 끝 지점
    targets = sorted(targets, key=lambda x: x[1])

    # 정렬된 타겟을 순회하며 요격 처리
    for start, end in targets:
        # 현재 요격 범위(shot)가 타겟의 범위 내에 있다면, 이미 요격한 것으로 간주
        if start < shot < end:
            continue  # 이미 요격했으므로 다음 타겟으로 넘어감

        # 현재 타겟의 끝 지점에 근접한 위치로 요격 위치를 갱신
        shot = end - 0.1
        # 새로운 타겟을 요격했으므로 요격한 타겟의 개수를 증가
        count += 1

    # 요격한 타겟의 총 개수를 반환
    return count