[프로그래머스(Programmers)][Java,Python] (Lv1) 붕대 감기 (PCCP 기출문제 1번)

 

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

프로그래머스

 

1. Java

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

 

2. Python

def solution(bandage, health, attacks):
    answer = health
    t = bandage[0]
    x = bandage[1]
    y = bandage[2]
    for i in range(len(attacks)) :
        attack_time = attacks[i][0]
        answer -= attacks[i][1]
        if answer <= 0 :
            answer = -1
            break
        if i == len(attacks) - 1 :
            break
        next_attack_time = attacks[i+1][0]
        recovery_time = next_attack_time - attack_time - 1
        answer = min( health, answer + (recovery_time * x) )
        if attack_time + t < next_attack_time :
            answer = min( health, answer + ( recovery_time // t * y ) )
    return answer