프로그래머스 2016년 Python

정답

def solution(a, b):
    answer = ''
    month = {1:31, 2:29, 3:31, 4:30, 5:31, 6:30, 7:31, 8:31, 9:30, 10:31, 11:30, 12:31}
    date = {0:'THU', 1:'FRI', 2:'SAT', 3:'SUN', 4:'MON', 5:'TUE', 6:'WED'}
    day = 0
    for i in range(1, a):
        day += month[i]
    day += b    
    answer += date[day%7]
    
    return answer

2016년 1월 1일이 금요일 임으로 date에 딕셔너리로 요일을 입력해주고 a와 b를 이용하여 총 날짜를 구한 후 7로 나눈 나머지로 요일을 구했습니다.