본문 바로가기
언어

2016

by 땅호720 2023. 12. 15.

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

'언어' 카테고리의 다른 글

해시  (1) 2023.12.18
카드 뭉치  (0) 2023.12.15
콜라  (0) 2023.12.15
명예의 전당  (0) 2023.12.15
leetcode (1)  (1) 2023.12.15