[프로그래머스] 코딩테스트 연습 > 연습문제 > 2016년

 

 

내가 맞힌 정답::

def solution(a, b):
    answer = ''
    totday = -1
    remday = 0
    
    days = {1:31, 2:29, 3:31, 4:30, 5:31, 6:30, 7:31, 8:31, 9:30, 10:31, 11:30, 12:31}
    
    #if a != 1:
    if a>1:
        for i in range(1, a):
            totday += days[i]
        totday += b
    elif a == 1:
        totday += b

    
    remday = totday%7
    
    if remday == 4:
        answer = 'TUE'
    elif remday == 5:
        answer = 'WED'
    elif remday == 6:
        answer = 'THU'
    elif remday == 0:
        answer = 'FRI'
    elif remday == 1:
        answer = 'SAT'
    elif remday == 2:
        answer = 'SUN'
    elif remday == 3:
        answer = 'MON'
    
    print(totday)
    
    return answer

 

https://school.programmers.co.kr/learn/courses/30/lessons/12901#

프로그래머스