[Python] 프로그래머스 / 붕대 감기

문제 링크

단순한 구현 문제이다.

 

정답 코드

from collections import deque

def solution(bandage, health, attacks):
    #       [5, 1, 5], 30,    [[2, 10], [9, 15], [10, 5], [11, 5]]
            #시전 시간, 초당 회복량, 추가 회복량
    
    T = deque(map(lambda x : x[0], attacks))
    P = deque(map(lambda x : x[1], attacks))
    fht = bandage[0]
    hps = bandage[1]
    sh = bandage[2]

    conti = 0
    blood = health
    
    for t in range(1, T[-1] + 1):
        #print(t, blood)
        if t in T :
            t_ = T.popleft()
            p_ = P.popleft()
            blood -= p_
            conti = 0
            
            if blood <= 0 :
                return -1 
            
            if not T :
                return blood
                
        else :
            conti += 1
            blood += hps
            
            if conti == fht :
                conti = 0
                blood += sh
                
            if blood > health :
                blood = health