본문 바로가기

내일배움 캠프/TIL

코딩테스트 2016년

def solution(a, b):
    yo_il = {0:"THU",1:"FRI",2:"SAT",3:"SUN",4:"MON",5:"TUE",6:"WED"}
    date = [30,31]
    date2 = [31,30]
    tot_day = 0
    for month in range(1,a):
        if month <8:
            tot_day += date[month%2]
        else:
            tot_day += date2[month%2]
    if a>2:
        tot_day -=1
    tot_day+=b
    return yo_il[tot_day%7]

배열로 30 31 를 돌아가게해서 월수를 에 있는 날짜를 세고 윤년이기 때문에 3월부터는 1일빼서 총 달의 일수를 구하고 day를 더한다음에 7로 나눠서 무슨요일인지 구했다