[프로그래머스/Python] 2016년

문제

프로그래머스 - 2016년

해결 과정

• 윤년: 2월 29일까지 있고, 1년 366일
• 요일을 배열로, 매달의 마지막 날짜를 배열로
• 입력 날짜까지의 일수를 계산
  • 만약 5월 25일이라면 31 + 29 + 31 + 30 + 25
• 입력 날짜까지의 일수를 7로 나눠서 해당 날짜 요일의 인덱스를 구한다.
• return (해당 날짜 요일의 인덱스에서 1을 뺀 값)번째 요일

시행 착오

• 복잡하게 생각하지말자

풀이

def solution(a, b):
    day = ['FRI','SAT','SUN','MON','TUE','WED','THU']
    end_date = [31,29,31,30,31,30,31,31,30,31,30,31]
    
    # 입력 날짜까지의 일수 계산
    n = sum(end_date[:a-1]) + b
    
    # 요일 구하기
    idx = n % 7

    return day[idx-1]