[프로그래머스] 에어컨(DP) LV3

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

프로그래머스

너무너무 어려웠는데 GPT 선생님도 명쾌한 답을 주시지 않았다 문제 해설보고 풀었는데.. 다음에 꼭 다시 풀어보기🤙

def solution(temperature, t1, t2, a, b, onboard):

    k = 1000*100
    t1 += 10
    t2 += 10
    temperature += 10
    
    dp = [[k for _ in range(51)] for _ in range(len(onboard))]
    # 초기값
    dp[0][temperature] = 0
    
    flag = 0
    # 조건: temperature는 t1~t2 범위 밖의 값이다
    if temperature > t2:
        flag = -1
    elif temperature < t1:
        flag = 1
    
    for i in range(1,len(onboard)):
        for j in range(51):
            temp = [k]
            if (onboard[i] == 1 and t1 <= j <= t2) or onboard[i] == 0:
                # 에어컨 ON 희망온도 (!=)
                if 0 <= j-flag <=50:
                    temp.append(dp[i-1][j-flag] + a)
                # 에어컨 ON 희망온도 (==)
                if t1 <= j <= t2:
                    temp.append(dp[i-1][j] + b)
                # 에어컨 OFF 실외온도 (!=)
                if 0 <= j+flag <=50:
                    temp.append(dp[i-1][j+flag])
                # 에어컨 OFF 실외온도 (==)
                if j == temperature:
                    temp.append(dp[i-1][j])
                
                dp[i][j] = min(temp)
    
    answer = min(dp[len(onboard)-1])
        
    return answer