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

문제

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

풀이

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

-1 해주기 귀찮아서 week 시작을 목요일로 바꿈

더 나은 풀이

import datetime

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

datetime 모듈 사용