[프로그래머스][1단계][python] 2016년 문제

문제 설명

2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까지 각각 SUN,MON,TUE,WED,THU,FRI,SAT 입니다.
예를 들어 a=5, b=24라면 5월 24일은 화요일이므로 문자열 "TUE"를 반환하세요.

제한 조건

2016년은 윤년입니다.
2016년 a월 b일은 실제로 있는 날입니다. (13월 26일이나 2월 45일같은 날짜는 주어지지 않습니다)

입출력 예

입력 : a = 5, b = 24
출력 : TUE

# 2016년은 윤년(2월29일)
def solution(a, b):
    # 월의 마지막 날짜 리스트(for에서 덧셈위해 0 임의로 0인덱스에 넣어줌)
    mon = [0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    # 요일 구할때 사용할 리스트(2016년 1월1일은 금요일. 따라서 0인덱스에 금요일부터 넣어줌)
    week = ['FRI','SAT','SUN','MON','TUE','WED','THU']

    # a-1월까지 마지막 날짜 더하기(ex.a가 3이면 mon[2]까지 더하기)
    sumMon = 0
    for i in range(1, a):
        sumMon += mon[i]
        
    day = sumMon + b # sumMon(a-1월까지 총합)에 b일더하기 그럼 a월b일까지의 총 일 나옴
    weekNumber = day%7 # day 나누기 7한 것의 나머지

    answer = week[weekNumber-1]
    
    return answer