[Programmers][Python][Level 1] 2016년

프로그래머스의 Level 01 2016년 문제입니다.

 

코딩테스트 연습 - 2016년

 

문제 

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"

 

풀이

저는 month라는 숫자 리스트에 각각의 달들의 일(day)을 기입하고 입력인 a(month), b(day)를 모두 더해서 나머지 연산 후 인덱스로 day 문자열 리스트에 접근해서 해결했습니다.

def solution(a, b):
    answer = ''
    
    # 2016년 1월 1일은 금요일
    day_arr = ["FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"]
    # 1,3,5,7,8,10,12 => 31일
    # 2 => 윤년 o 29일, 윤년 x 28일
    # 4,6,9,11 => 30일
    month_arr = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    day = 0
    
    for m in range(a-1):
        day += month_arr[m]
    
    day += b
    day %= 7 
    
    answer = day_arr[day-1]
    
    return answer

 

코드 정리

기존 for문의 역할은 사용자 입력한 a(month)가 숫자 5라면 5(1~4)까지 더해주는 것이기에 sum()을 이용해서 해결했습니다.

def solution(a, b):
    answer = ''
    
    # 2016년 1월 1일은 금요일
    day_arr = ["FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"]
    month_arr = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    day = 0
    
    day = sum(month_arr[:a-1]) + b - 1
   
    answer = day_arr[day % 7]
    
    return answer

 

다른 사람의 풀이

다른 분의 풀이를 봤는데, 역시 라이브러리를 잘 사용하는 것도 중요한 것을 느낍니다.

date를 이용해서 해당 날짜를 문자열로 변환하는 strftime('%a')를 이용하고 upper()를 이용해서 대문자로 변경해서 해결이 가능합니다... 와우

from datetime import date

def solution(a, b):
    
    return date(2016,a,b).strftime('%a').upper()