[프로그래머스] 2016년

Question

def solution(a, b):
    from datetime import date
    Month = a
    Day = b
    week = ['MON','TUE','WED','THU','FRI','SAT','SUN']
    result = week[date(2016,Month,Day).weekday()]
    return result

 

다른 사람들의 풀이를 보니 사전을 이용하는 방법도 있다.

def getDayName(a,b):
    day_name = ['THU', 'FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED']
    month_dict = {
        "1":31, 
        "2":29, 
        "3":31, 
        "4":30, 
        "5":31, 
        "6":30, 
        "7":31, 
        "8":31, 
        "9":30, 
        "10":31, 
        "11":30, 
        "12":31
    }
    days = 0
    for i in range(1, a):
        days += month_dict[str(i)]
    days += b
    index = days % 7

    return day_name[index]