루나의 TIL 기술 블로그

프로그래머스 level 1 2016년

14 Jan 2022 | 기초 프로그래머스 TIL

걸린 시간 : 1시간

문제 설명

2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까지 각각 SUN,MON,TUE,WED,THU,FRI,SAT

입니다. 예를 들어 a=5, b=24라면 5월 24일은 화요일이므로 문자열 “TUE”를 반환하세요.

제한 조건

2016년은 윤년입니다.
2016년 a월 b일은 실제로 있는 날입니다. (13월 26일이나 2월 45일같은 날짜는 주어지지 않습니다)

입출력 예

a b result
5 24 "TUE"

제출 코드

def solution(a, b):
    day_dic = {"1":"FRI", "2":"SAT", "3":"SUN", "4":"MON", "5":"TUE", "6":"WED", "0":"THU"}
    #2016년 1월 1일이 금요일이니까
    daylist = [31,29,31,30,31,30,31,31,30,31,30,31]
    #윤년이라 2월은 29일까지
    day=0
    
    if a != 0:
        for m in range(a-1):
            day += daylist[m]
    
    day = day + b
    answer = str(day%7)

    for key, value in day_dic.items():
        answer = answer.replace(key, value)
    return answer
    #숫자를 문자로 바꾸는 부분은 저번에 풀었던 '숫자 문자열과 영단어'에서 가져왔다

키밸류 replace는 문자열만 되는 모양이다.

String.prototype.replace() : replace() 메서드는 어떤 패턴에 일치하는 일부 또는 모든 부분이 교체된 새로운 문자열을 반환합니다. 그 패턴은 문자열이나 정규식(RegExp)이 될 수 있으며, 교체 문자열은 문자열이나 모든 매치에 대해서 호출된 함수일 수 있습니다. - MDN

모범 답안

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]

sum(months[:a-1]) 아아주 간단명료하다

손가락 코딩

프로그래머스에 이런 코드가 박제되어있었다…. ㅋㅋㅋ

def getDayName(a,b):
    answer = ""
    if a>=2:
        b+=31
        if a>=3:
            b+=29#2월
            if a>=4:
                b+=31#3월
                if a>=5:
                    b+=30#4월
                    if a>=6:
                        b+=31#5월
                        if a>=7:
                            b+=30#6월
                            if a>=8:
                                b+=31#7월
                                if a>=9:
                                    b+=31#8월
                                    if a>=10:
                                        b+=30#9월
                                        if a>=11:
                                            b+=31#10월
                                            if a==12:
                                                b+=30#11월
    b=b%7

    if b==1:answer="FRI"
    elif b==2:answer="SAT" 
    elif b==3:answer="SUN"
    elif b==4:answer="MON"
    elif b==5:answer="TUE"
    elif b==6:answer="WED"
    else:answer="THU"
    return answer


print(getDayName(5,24))

내장 함수 코딩

from datetime import date
DAY = {0: 'MON', 1: 'TUE', 2: 'WED', 3: 'THU', 4: 'FRI', 5: 'SAT', 6: 'SUN'}

def getDayName(a,b):
    d = date(2016, a, b)
    return DAY[d.weekday()]

빠른 길을 좋아하는 사람들을 위한 내장함수 코딩도 있다.

Related Posts