코딩테스트/Python
[프로그래머스] 2016년
yo~og
2021. 8. 3. 16:20
내 코드
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의 나머지를 구했다.