programmers | Lv1. 2016년 [Python]

💡 Python 3

🔗 문제

2016년 [Link]

💻 코드

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

💥 다른 사람 코드

datatime 라이브러리 활용

import datetime

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