[프로그래머스] 2016년 (Python)

문제 링크 (Level 1)

https://school.programmers.co.kr/learn/courses/30/lessons/12901?language=python3 

프로그래머스

 

나의 코드 (Python)

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

 

for를 a(월)만큼 돌려서 그 합과 b(일)을 더한 값을 % 7로 해주면 무슨 요일인지 알 수 있다.

weekday 배열은 1월 1일이 'FRI'이므로 ['THU','FRI','SAT','SUN','MON','TUE','WED']순으로 배열하면 된다.