[Algorithm🧬] 2016년

문제 / 풀이.py

def solution(a, b):
    answer = ''
    month = [0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    day = ["MON", "TUE", "WED", "THU", "FRI", "SAT", "SUN"]

    # 이전 월까지 날짜 + 주어진 날짜 를 해서 5월 1일을 1월 122일로 보고 7로 나눈 나머지를 기준으로 요일을 반환한다.
    # 날짜 계산에서 요일을 찾으려면 (끝나는 날짜 - 시작 날짜) % 7 을 해야함.
    JanFirst = 3 # 월요일이 0 이라 금요일은 4 인데 미리 -1 (1월 1일) 해준다.
    sum = JanFirst + b 
    
    for i in range(a):
        sum += month[i]

    return day[sum % 7]