[프로그래머스, 파이썬] 2016년

풀이

• 2016년도의 특정 일에 대한 요일을 구하는 문제라서 
• 1월1일이 금요일이라고 하여 찾아보니 실제로 맞는 요일이었다.
• 만약 다를 경우 연산을 조금 더 추가해서 적용하면 되는 것이기에 큰 문제는 없지만
• 같기에 수월했다.
• 그래서 파이썬 함수인 datetime을 가져온 다음
• int 타입인 연도, 월, 일을 넣어주고 .weekday()+1 함수를 적용하게 되면
• 현재 요일에 대한 숫자가 return되기에 요일에 대한 배열을 만들어 출력하여 해결

 

from datetime import datetime

def solution(a, b):

    weekend = ["SUN","MON","TUE","WED","THU","FRI","SAT"]

    return weekend[(datetime(2016,a,b).weekday()+1)%7]