Algorithm/프로그래머스

[Python] 2016년

내영잉 2021. 5. 1. 23:33

1. 문제 📚

2. 입출력 예 📋

3. 알고리즘 ✅

  • 각 달의 말일을 담은 리스트를 만든다
  • 1월 1일이 FRI이므로, FRI부터 시작하는 요일 리스트를 만든다.
  • a-1달까지의 말일을 더한 후, b-1을 더하고 %7번째의 요일을 리턴한다.
    -> 1월1일이 FRI이므로 1월 1일이 0번째를 만들기 위해 각각 -1을 한다.

4. 소스코드 💻

def solution(a,b):
    months = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    days = ['FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED', 'THU']
    return days[(sum(months[:a-1])+b-1)%7]