[Python] 2016년

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

프로그래머스

# 2016년
def solution(a, b):

    month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    days_array = ['THU', 'FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED']

    days = 0

    for i in range(a-1):
        days += month[i]

    days  = days + b 
    week_day = days % 7
    answer = days_array[week_day]
    
    return answer

 

• 또 다른 풀이

# 또 다른 풀이
import datetime

def solution(a,b):
    t = 'MON TUE WED THU FRI SAT SUN'.split()
    return t[datetime.datetime(2016, a, b).weekday()]