[프로그래머스 / lv.1] 2016년

in Problem Solving on Programers


날짜: 2021년 5월 16일
소요 시간: 1분 28초
카테고리: 수학문제
태그: 레벨1, 파이썬

코딩테스트 연습 - 2016년

입출력 예시

abresult
524“TUE”

내가 적은 코드

import datetime

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

풀이 과정

datetime 라이브러리를 가져와서 2016년 a월 b일의 요일을 출력했다.

베스트 코드

def getDayName(a,b):
    months = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    days = ['FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED', 'THU']
    return days[(sum(months[:a-1])+b-1)%7]

반성

  • 코드를 짧게하기 위해 datetime 모듈을 들고왔다.
  • 모듈을 굳이 들고오지않아도 수식으로 나와 시간복의 코드를 짠 사람이 있었다.


© 2021. by hminkim