IT보안관의 공부 클라우드

[프로그래머스 Level1]2016년 본문

코딩 테스트/프로그래머스

[프로그래머스 Level1]2016년

ㅡㅡㅡㅡㄷ 2022. 6. 28. 21:30

코딩테스트 연습 - 2016년 | 프로그래머스 (programmers.co.kr)

 

코딩테스트 연습 - 2016년

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

programmers.co.kr

week = ['FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED', 'THU']
months = {1: {}, 2: {}, 3: {}, 4: {}, 5: {}, 6: {}, 7: {}, 8: {}, 9: {}, 10: {}, 11: {}, 12: {}}
days = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31] #윤년은 2월이 29일까지
chk = 0
for i in range(1, 13):
    month = {}
    for j in range(1, days[i-1] + 1):
        if chk > 6:
            chk = 0
        month[j] = week[chk]
        chk+=1
    months[i] = month
    print(months)


def solution(a, b):
    answer = months[a][b]
    return answer
Comments