프로그래머스. 2016년 파이썬 풀이

프로그래머스 연습문제. Level 1. 2016년 파이썬 풀이

문제링크 https://programmers.co.kr/learn/courses/30/lessons/12901

2016년은 윤년이니 2월은 29일까지 있다
입력받은 월 전까지의 날짜를 모두 더하고
입력받은 일에 1을 뺀 값을 더한다.
2016년이 금요일부터 시작했으니 +5하고 그 값을 %7하여 요일을 확인한다.

def solution(a, b):
    answer = ''
 
    month = [ 31,29,31,30,31,30,31,31,30,31,30,31 ]
    week = [ "SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT" ];
    
    day = 0
    
    for i in range(a-1):
        day += month[i]
        
    day += b-1
    
    answer_day = (5 + day) % 7
    
    return week[answer_day]