알고리즘 연습/기초 (입출력 등)

[Lv.1 / 프로그래머스 / 파이썬] 2016년

김세진 2024. 4. 12. 15:18
반응형

 

 

 

 

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

 

풀이

 

파이썬에서 날짜를 다루는 라이브러리인 datetime을 사용해 간단하게 해결할 수 있다.

datetime 객체에 weekday() 함수를 사용해 현재 날짜의 요일을 반환받을 수 있다.

단, 인덱스 0이 월요일임에 주의하자.

 

import datetime

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

 

 

 

 

 

반응형