(Python) 2016년 (프로그래머스 Lv.1)

2016년 (프로그래머스 Lv.1)

https://school.programmers.co.kr/learn/courses/30/lessons/12901

프로그래머스

내 코드

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 #b를 그대로 더해야 하니까 0부터 시작
    for i in range(a-1):#예를 들어 3월은 2월까지의 일자만 더 해야함
        day += month[i] 
    day += b #남은 일 수 더해주기
    answer = week[(5+(day % 7))%7-1] #처음 0부터 시작했으니 금요일 전인 5더해주기,7로 나눈 나머지
    return answer

각 달의 일 수와 일주일의 문자열을 배열로 만들어 정리했다.

코드를 짜면서 상수에 관해 헷갈리는 부분들이 있었는데 위 코드에 주석으로 정리해 두었다.