언어

2016

땅호720 2023. 12. 15. 18:06

1월 1일은 금요일

리스트 인덱스가 0부터니까 금요일부터 시작 (당일)

def solution(a, b):
    month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    week = ['FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED', 'THU']
    
    if a==1:
        answer = week[b%7-1]
    else:
        # 5월이면 4월까지 더하고 남은 일수 진행
        day = sum(month[:a-1]) + b
        answer = week[day%7-1]

    return answer