[PCCP 기출문제] 1번 / 붕대 감기

문제

https://school.programmers.co.kr/learn/courses/19344/lessons/242258

프로그래머스

 

---

풀이

def solution(bandage, health, attacks):
    max_health = health
    t, x, y = bandage
    end_time = attacks[-1][0]
    cur_t = 0
    cur_health = health

    attacks_dic = dict(attacks)

    for i in range(end_time + 1):
        # 공격
        if i in attacks_dic:
            cur_t = 0
            cur_health -= attacks_dic[i]

            if cur_health <= 0:
                return -1
            continue

        # 공격 받지 않았을 때
        cur_t += 1
        cur_health += x

        # 추가 회복
        if cur_t == t:
            cur_health += y
            cur_t = 0

        cur_health = min(cur_health, max_health)
    return cur_health

---

설명

주어진 조건대로 시뮬레이션 하면서 결과를 return 하는 문제 라고 한다.

아 근데 제대로 문제를 이해하지 못해서 다른 사람의 코드를 참조했다.

 

1. attacks 배열을 딕셔너리로 변형해서 시간복잡도를 확보하자!

2. 공격 받았을 때를 먼저 확인 해야 정답이 됨. 공격 받지 않았거나, 추가 회복일때를 먼저 확인하니 답이 +2 가 됐었었음.