파이선 코딩 테스트[0] LV1 - 붕대 감기-프로그래머스 Python

 

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

프로그래머스

 

문제 설명
어떤 게임에는 붕대 감기라는 기술이 있습니다.

붕대 감기는 t초 동안 붕대를 감으면서 1초마다 x만큼의 체력을 회복합니다. t초 연속으로 붕대를 감는 데 성공한다면 y만큼의 체력을 추가로 회복합니다. 게임 캐릭터에는 최대 체력이 존재해 현재 체력이 최대 체력보다 커지는 것은 불가능합니다.

기술을 쓰는 도중 몬스터에게 공격을 당하면 기술이 취소되고, 공격을 당하는 순간에는 체력을 회복할 수 없습니다. 몬스터에게 공격당해 기술이 취소당하거나 기술이 끝나면 그 즉시 붕대 감기를 다시 사용하며, 연속 성공 시간이 0으로 초기화됩니다.

몬스터의 공격을 받으면 정해진 피해량만큼 현재 체력이 줄어듭니다. 이때, 현재 체력이 0 이하가 되면 캐릭터가 죽으며 더 이상 체력을 회복할 수 없습니다.

당신은 붕대감기 기술의 정보, 캐릭터가 가진 최대 체력과 몬스터의 공격 패턴이 주어질 때 캐릭터가 끝까지 생존할 수 있는지 궁금합니다.

붕대 감기 기술의 시전 시간, 1초당 회복량, 추가 회복량을 담은 1차원 정수 배열 bandage와 최대 체력을 의미하는 정수 health, 몬스터의 공격 시간과 피해량을 담은 2차원 정수 배열 attacks가 매개변수로 주어집니다. 모든 공격이 끝난 직후 남은 체력을 return 하도록 solution 함수를 완성해 주세요. 만약 몬스터의 공격을 받고 캐릭터의 체력이 0 이하가 되어 죽는다면 -1을 return 해주세요.

 

문제 분석

1. attacks의 공격 시간중 가장 큰 값을 파악해서, 순환을 돌린다.

2. 최대 hp 값을 기준으로 값을 제한한다

3. attack에 따라 회복 유무를 결정한다.

4. 추가 회복 쿨타임을 체크해서 회복시킨다.

5. 문제의 예시에서 표를 보고 표에 있는 값을 활용 하면 쉽다.

 

정답 ( 살짝 하드 코딩 느낌)

def solution(bandage, health, attacks):
    bandage_time, bandage_sec, bandage_add = bandage    
    hp = health
    attacks_time = 1
    for i in attacks:
       attacks_time = max(attacks_time, i[0]) 

    bandage_cnt = 0
    attacks_cnt = 0
    for time in range(1, attacks_time+1):
        for x in attacks:
            if time == x[0]:
                hp -= x[1]
                bandage_cnt = 0
                if hp <= 0:
                    return -1
                attacks_cnt = 1

        if attacks_cnt == 0:                
            hp += bandage_sec
            bandage_cnt += 1
            if bandage_cnt == bandage_time:
                hp += bandage_add
                bandage_cnt = 0
        attacks_cnt = 0
        hp = min(hp, health)
        print(time, hp)
    answer = hp
    return answer

 

사용 기능

min()

max()

 

 

깔끔하게 정리된 수식 

 

def solution(bandage, health, attacks):
    currentHealth = health  # 현재 체력
    success = 0             # 연속 성공
    attack = False
    bandageTime, recoveryBySecond, additionalRecovery = bandage

    attacksSecond = [ attack[0] for attack in attacks ]     # 공격을 받은 시간

    print(f"{0} {currentHealth} {success} {attack}")
    for second in range(1, attacksSecond[-1]+1):
        if second in attacksSecond:     # 공격 받음
            attack = True
            success = 0     # 기술 취소
            currentHealth -= attacks[attacksSecond.index(second)][1]    # 공격 피해
        else:
            attack = False
            currentHealth += recoveryBySecond   # 초당 회복
            success += 1
            if success >= bandageTime:          # 연속 성공이면 추가 회복
                currentHealth += additionalRecovery
                success = 0
            if currentHealth > health:          # 최대 체력 보정
                currentHealth = health
        print(f"{second} {currentHealth} {success} {attack}")
        if currentHealth <= 0:
            return -1
    return currentHealth

사용기능

리스트의 끝요소 확인

attacksSecond[-1]

리스트 컴프리핸션for문

attacksSecond = [ attack[0] for attack in attacks ]   

.index 해당 인덱스 값 추출

currentHealth -= attacks[attacksSecond.index(second)][1] 

가장 짧은 코드

def solution(bandage, health, attacks):
    hp = health
    start = 1
    for i, j in attacks:
        hp += ((i - start) // bandage[0]) * bandage[2] + (i - start) * bandage[1]
        start = i + 1
        if hp >= health:
            hp = health
        hp -= j
        if hp <= 0:
            return -1
    return hp

이건 매우 간결하고 이해하기 쉽지만, 과연 시험에서 쓸수 있을진 모르겠다.