[알고리즘] 프로그래머스 1단계 - 2016년

def solution(a, b):
    
    week = ['FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED', 'THU']
    month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    
    return week[(sum(month[0:a-1])+b-1)%7]

풀이과정

1. 요일과 달을 저장할 배열 week와 month를 만든다.

   • 2016년 1월 1일은 금요일으로 'FRI'부터 시작한다.
   • 윤년은 2월 29일이 있는 년도이다.

2. a-1 달까지 더한 후, b-1 일을 더하고, 7로 나눈 인덱스 값이 답이다.