[프로그래머스] / [Level 2] / [Python] 디펜스 게임

https://school.programmers.co.kr/learn/courses/30/lessons/142085

프로그래머스

---

• 무적권 스킬은 heap에 저장되어 있는 원소에 사용한다.
• 적이 많은 라운드에서 무적권 스킬을 사용해야한다.
• 반대로 적이 적은 라운드에선 heappop을 사용하여 남은 병사 n과 겨룬다.

from heapq import heappush, heappop

def solution(n, k, enemy):
    answer = 0
    
    heap = []
    
    for idx, e in enumerate(enemy):
        heappush(heap, e)
        
        if len(heap) > k:
            n -= heappop(heap)
            if n < 0:
                answer = idx
                break
            elif n == 0:
                answer = idx + 1
                break
    else:
        answer = idx + 1
            
    return answer