본문 바로가기

코딩테스트/Python

[프로그래머스] 2016년

내 코드

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

먼저 월을 다 더해주고 일을 더해주었다. 1월 1일은 빼야하므로 1을 빼주었다.

요일을 넣은 리스트 day를 만들었다.

1월 1일이 금요일이므로 sum에 5를 더한뒤 7의 나머지를 구했다.