[프로그래머스 - lv1] 2016년

\

 

문제 링크 : https://school.programmers.co.kr/learn/courses/30/lessons/12901

프로그래머스

---

1) 풀이

파이썬의 calander 모듈을 이용하면 편리하게 해결할수 있다.

단, 문제에서는 요일 순서가 일 ~ 토로 되어있는데, 이걸 월~일로만 변경해주면 된다.

파이썬 공식 홈페이지에 calander.weekday() 함수에 대한 설명이 다음과 같이 나와있다.

https://docs.python.org/ko/3/library/datetime.html?highlight=weekday 

datetime — Basic date and time types

이 함수를 이용해서 구현하면 다음과 같다.

import calendar

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

미리 요일 list를 선언해 둔 후에 weekday 함수로부터 리턴받는 인덱스값에 해당하는 데이터를 return해주면 끝.

 

2) 결과