[프로그래머스 Python] Lv 1. 2016년

---

문제 설명

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일같은 날짜는 주어지지 않습니다)
• 문제 보러가기

문제 풀이

(1) Pseudo-Code

• datetime 라이브러리를 불러와 주어진 날짜에 대한 요일 번호를 불러온다.
• 출력해야하는 요일(문자열 형태)를 요일 번호에 맞는 위치에 담은 list를 만든다.
• 문자열 형태의 요일이 담긴 리스트에 요일 번호(idx) 값으로 문자열 형태의 요일을 반환한다.

(2) 코드 작성

from datetime import datetime

def solution(a, b):
    idx = datetime(2016, a, b).weekday()
    weeks = ['MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT', 'SUN']
    return weeks[idx]

👩🏻‍💻개인 공부 기록용 블로그입니다
오류나 틀린 부분이 있을 경우 댓글 혹은 메일로 따끔하게 지적해주시면 감사하겠습니다.