[프로그래머스] level1 - 2016년 : 파이썬(Python) 풀이

문제

프로그래머스 level1 2016년

코딩테스트 연습 - 2016년

 

코드

import datetime
def solution(a, b):
    week = {0:"MON", 1:"TUE", 2:"WED",3:"THU", 4:"FRI", 5:"SAT", 6:"SUN"}
    return week[datetime.date(2016,a,b).weekday()]

 

 

코드풀이

문제는 간단하다.

a, b 를 입력받아서 2016년 a 월 b 일이 무슨 요일인지 출력하면 되는 문제이다.datetime 모듈을 통해서 날짜를 입력받아 weekday( ) 함수를 통해 요일에 대한 정수를 반환하였고, 미리 숫자에 맞는 요일을 딕셔너리 형태로 만들어 value 값을 리턴하였다.