65-9kyo-hwang

[9kyo-hwang]

🔗 문제 링크

디펜스 게임

✔️ 소요된 시간

1시간

✨ 수도 코드

최대 $1,000,000$ 라운드에 걸쳐 적들이 들어온다. 그리고 최대 $500,000$라운드를 무적권을 사용해 병사 소모를 0으로 만들 수 있다.

백트래킹 등을 이용해 모든 경우의 수를 구하기에는 숫자가 너무 크다. 라운드가 최대 $1,000,000$이기 때문에 $O(N^2)$ 풀이법도 불가능하다. 즉 $O(N logN)$ 풀이법을 사용해야 한다.

직관적으로 생각했을 때는, 무적권은 가능한 "적의 수가 많은 라운드"에 사용하는 것이 좋다. 그러나 이러한 방법은 "선형탐색"으론 불가능하고, 그렇다고 위에서 언급한 것처럼 모든 경우의 수를 다 확인할 수는 없다.

따라서 여기서는 우선순위 큐를 사용한다. 원리는 간단하다.

1. 일단은 매 라운드마다 등장하는 적 수에 맞게 병사를 소모한다. 그리고 동시에 적 수를 MaxHeap에 기록한다.
2. 만약 남은 병사가 존재한다면 이어서 라운드를 진행한다.
3. 남은 병사가 하나도 없다면 아래 로직을 수행한다.
   i. 만약 더 이상 무적권을 사용할 수 없다면 라운드를 중단한다.
   ii. 아니라면 MaxHeap에서 지난 라운드 중 가장 적이 많았던 라운드의 적 수를 pop해 그 수 만큼 병사 수를 복원시키고 남은 무적권 횟수를 1 차감시킨다.

이런 식으로 MaxHeap을 사용하면 $O(logN)$ 시간복잡도로 지난 라운드 중 가장 적이 많은 라운드 정보를 얻어올 수 있게 된다.

#include <string>
#include <vector>
#include <queue>

using namespace std;

int solution(int N, int K, vector<int> InEnemies) 
{
    priority_queue<int> NumofEnemiesBlocked;  // MaxHeap
    int Round = 0;
    
    for(int Enemy : InEnemies)
    {
        N -= Enemy;
        NumofEnemiesBlocked.emplace(Enemy);
        
        if(N < 0)  // 무적권을 사용해야 함!
        {
            if(K == 0)  // 더 이상 라운드 진행 불가
            {
                break;
            }
            
            N += NumofEnemiesBlocked.top();  // 이전 라운드 중 가장 적이 많았던 라운드 정보 추출
            NumofEnemiesBlocked.pop();
            K--;
        }
        
        Round++;
    }
    
    return Round;
}

📚 새롭게 알게된 내용

왤케 그리디 유형은 안풀리는지 모르겠다.
예전에도 코테볼 때 그리디/우선순위큐 쓰는 문제 못 풀었었는데, 이것도 한참 삽질하다가 겨우 해결했다 흠...

[mjj111]

문제 다 풀고, 교황님 코드 보니 round 변수명이 더 명확해서 바로 냅다 수정했습니다 ㅋㅋㅋㅋ

import heapq 
def solution(n, k, enemy):
    answer = len(enemy)
    pq = []
    my = n
    card = k

    for round in range(len(enemy)):
        my -= enemy[round]
        heapq.heappush(pq, -enemy[round])
        if my < 0:
            if card > 0:
                my += heapq.heappop(pq) * -1 
                card -= 1
            else :
                return round
    return answer

[gjsk132]

오랜만에 보는 최대힙 슥삭~

from heapq import *

def solution(n, k, enemy):
    round = 0

    fight = []

    for e in enemy:
        round += 1
        n -= e
        heappush(fight, -e)

        while n < 0 and k > 0:
            k -= 1
            n -= heappop(fight)
        
        if n < 0 :
            round -= 1
            break

    return round