[프로그래머스] 2016년

 

 

 

안녕하세요~

27년차 진로탐색꾼 조녁입니다!!

1일 1코테 10일차!!! 하루에 한문제씩 풀면 안될거같아서 오늘은 세문제까지 풀어봅니다. 

 

기본 연습문제로 난이도는 쉬웠습니다.

 

· 문제명 : 2016년

· 사용 언어 : Python3

· 난이도 : Level1 

· 문제 설명 

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):
	#윤년의 월별 전체 날짜수 딕셔너리로 저장 
    months ={1:31,2:29,3:31,4:30,5:31,6:30,7:31,8:31,9:30,10:31,11:30,12:31}
    
    #1월1일이 금요일이었으므로 금요일부터해서 반복되는 7일 리스트
    days = ["FRI","SAT","SUN","MON","TUE","WED","THU"]
    
    #1일~ b일까지 b-1번 요일이 변하므로 b-1로 NOW 할당
    NOW = b-1
    
    #NOW(1월1일부터 오늘까지 +된 일수)를 7로 나눈 나머지를 구해준다.
    for i in range(1,a):
        NOW += months[i]
        
    answer = days[NOW%7]
    return answer