[프로그래머스/파이썬] 2016년

-코드

def solution(a, b):
    mo = [31,29,31,30,31,30,31,31,30,31,30,31]
    r = 0
    for i in range(a - 1):
        r += mo[i]
    r += b
    if r % 7 == 1:
        ans = "FRI"
    elif r % 7 == 2:
        ans = "SAT"
    elif r % 7 == 3:
        ans = "SUN"
    elif r % 7 == 4:
        ans = "MON"
    elif r % 7 == 5:
        ans = "TUE"
    elif r % 7 == 6:
        ans = "WED"
    elif r % 7 == 0:
        ans = "THU"
    return ans