Programmers / [PCCP 모의고사 1회] 3번 / 붕대 감기 / python

 

 

문제 : https://school.programmers.co.kr/learn/courses/30/lessons/250137?language=python3

 

나의 풀이

def attack(health, damage, max_health):
    health -= damage
    if health <= 0:
        return -1, 0
    return health, 0

def recover(health, count, bandage, max_health):
    health += bandage[1]
    count += 1

    if count == bandage[0]:
        health += bandage[2]
        count = 0

    health = min(health, max_health)
    return health, count

def solution(bandage, health, attacks):
    max_health = health
    count = 0
    attack_count = 0
    time = 1

    while time <= attacks[-1][0]:
        if attack_count < len(attacks) and attacks[attack_count][0] == time:
            health, count = attack(health, attacks[attack_count][1], max_health)
            attack_count += 1
            
            if health == -1:
                return -1
        else:
            health, count = recover(health, count, bandage, max_health)

        time += 1

    return health

 

 

+ 함수 하나로 표현한 더 간단한 풀이

def solution(bandage, health, attacks):
    max_health = health
    count = 0
    attack_count = 0
    time = 1
    
    while time <= attacks[-1][0]:
        if attack_count < len(attacks) and time == attacks[attack_count][0]:
            health -= attacks[attack_count][1]
            count = 0
            attack_count += 1
            
            if health <= 0:
                return -1
        else:
            health += bandage[1]
            count += 1
        
            if count == bandage[0]:
                health += bandage[2]
                count = 0
            
            health = min(health, max_health)
            
        time += 1
    return health