붕대 감기

문제 링크

문제 풀이 시간 :

문제 요약

• bandage는 [시전 시간 t, 초당 회복량 x, 추가 회복량 y]로 이루어진 배열
• t초 연속으로 붕대를 감는 데 성공하면 y만큼 체력을 추가로 회복
• 몬스터의 공격을 받으면 연속 성공이 초기화되고, 공격당하는 순간에는 회복할 수 없음
• 체력은 최대 체력을 넘을 수 없고, 0 이하가 되면 -1을 반환

문제 풀이

처음엔 0초부터 마지막 공격 시각까지 1초 단위로 반복문을 돌면서 체력을 그대로 시뮬레이션하면 되겠다고 생각했다.

매초 회복량을 더하고 연속 성공 횟수를 세다가 t초를 채우면 추가 회복을 주고, attacks에 있는 시간이 되면 피해를 주는 방식이다.

하지만 이렇게 짜면 매초마다 공격 시각인지 확인해야 하고, 회복과 피해가 같은 초에 겹치는 경우("공격을 당하는 순간에는 회복할 수 없다")를 계속 신경 써서 분기를 나눠야 한다.

attacks의 길이는 최대 100이지만 공격 시간은 최대 1,000까지 갈 수 있어서, 초 단위로 도는 구현은 자잘한 예외 처리가 많아진다.

이 문제의 핵심은 공격과 공격 사이의 구간을 한 번에 계산하면 된다는 점이다.

attacks[i][0] - t - 1로 이전 공격 이후 흐른 시간을 구하고, 그 시간 동안 기본 회복(x * time)과 t초마다 발생하는 추가 회복(y * (time / t))을 한 번에 더한다.

그 다음 attacks[i][1]만큼 체력을 깎고, 0 이하가 되면 바로 -1을 반환한다.

이렇게 하면 초 단위 반복 없이 공격 횟수만큼만 순회해도 되고, 공격이 일어나는 순간에는 회복이 없다는 조건도 "공격 직전까지만 회복을 계산한다"는 방식으로 자연스럽게 만족된다.

최종 코드

class Solution {
    public int solution(int[] bandage, int health, int[][] attacks) {
        int answer = health;
        
        int t = 0;
        
        for(int i=0;i<attacks.length;i++){
            int time = attacks[i][0] - t -1;
            int temp = answer + bandage[1] * time;
            temp += (time / bandage[0]) * bandage[2];
            
            answer = Math.min(health, temp);
            
            answer -= attacks[i][1];
            
            if(answer <= 0)
                return -1;
            
            t = attacks[i][0];
        }
        
        return answer;
    }
}