알고리즘/프로그래머스

[프로그래머스] 2016 (python)

김홍중 2021. 5. 26. 22:51
def solution(a, b):
    day = ['SUN', 'MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT']
    all_day = 366
    year = []
    month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    target_day = 0
    
    for i in range(366):
        year.append(day[(i + 5) % 7])
    
    for i in range(a-1):
        target_day += month[i]
        
    target_day += b - 1
    return year[target_day]