문제 설명
2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까지 각각 SUN,MON,TUE,WED,THU,FRI,SAT
입니다. 예를 들어 a=5, b=24라면 5월 24일은 화요일이므로 문자열 TUE를 반환하세요.
제한조건
- 2016년은 윤년입니다.
- 2016년 a월 b일은 실제로 있는 날입니다. (13월 26일이나 2월 45일같은 날짜는 주어지지 않습니다)
입출력 예
a | b | result |
5 | 24 | 'TUE' |
나의 풀이
def solution(a, b):
y =[31,29,31,30,31,30,31,31,30,31,30,31 ]
n=['FRI','SAT','SUN','MON','TUE','WED','THU']
k=sum(y[:a-1]) + b-1
return n[k % 7]
각 요일의 이름과 달의 마지막 날짜를 리스트로 저장합니다.
a월 b일을 구하기 위해 a달 전까지의 날짜를 더하고 b일을 더해줍니다.
(1월 1일은 금요일이니까 하루 계산해주었기 때문에 -1 하기)
계산한 날짜로 요일의 수인 7을 나눠 값을 구합니다.
프로그래머스 '2016년' : programmers.co.kr/learn/courses/30/lessons/12901
'알고리즘 > 프로그래머스' 카테고리의 다른 글
[프로그래머스 Level 1] 같은 숫자는 싫어 - 파이썬(Python) (0) | 2020.06.02 |
---|---|
[프로그래머스 Level 1] 가운데 글자 가져오기 - 파이썬(Python) (0) | 2020.06.02 |
[프로그래머스 Level 1] K번째수 - 파이썬(Python) (0) | 2020.06.01 |
[프로그래머스 Level 1] 체육복 - 파이썬(Python) (0) | 2020.06.01 |
[프로그래머스 Level 1] 모의고사 - 파이썬(Python) (0) | 2020.06.01 |