[프로그래머스] Lv1. 2016년 (파이썬)

금요일을 0번째 요일로 보고 각 달마다 일 수와 b일까지의 일수를 더해준 다음에 7로 나눈다.

def solution(a, b):
    answer = ''
    k=["FRI","SAT","SUN",'MON','TUE','WED','THU']   #금요일부터 시작
    d=0 # 일 수
    for i in range(1,a):
        if i in [1,3,5,7,8,10]: #31일인 달
            d+=31
        elif i in[4,6,9,11]:    #30일인 달
            d+=30
        elif i == 2:    #29일인 달
            d+=29
    d+=b-1  #b일에서 1빼줌
    answer=k[d%7]   #일 수를 7로 나눈 나머지
    return answer