[프로그래머스] 2016년 - 파이썬

Programmers

2016년

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

프로그래머스

문제

코드

• 각 달(월)의 말일을 리스트로 만들기
• day 순서는 2016년 1월 1일이 금요일이기 때문에 금요일부터 정렬
• total: 1월 1일부터 a월 b일 총 일수
• day는 금요일부터 시작하기 때문에, % 7의 결과가 0이면 금요일, 1이면 토요일 등으로 매핑되어야 함. 하지만 Python에서 배열 인덱스는 0부터 시작하므로, 계산된 나머지에서 1을 빼주어 올바른 인덱스로 조정

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

    total = sum(months[:a-1]) + b
    return day[total % 7 - 1]