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]