Coding Test: 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”

import calendar

def solution(a, b):
    answer = ''
    day = 0

    day = calendar.weekday(2016,a,b)

    if day == 0:
        answer = 'MON'
    if day == 1:
        answer = 'TUE'
    if day == 2:
        answer = 'WED'
    if day == 3:
        answer = 'THU'
    if day == 4:
        answer = 'FRI'
    if day == 5:
        answer = 'SAT'
    if day == 6:
        answer = 'SUN'

    return answer

---

calendar.weekday()로 쉽게 해결!

---

번외

datetime을 import하여도 가능합니다 (아래)

import datetime
datetime.datetime(2016,5,23).weekday()

출제의도에 맞는(?) 풀이법

이전 포스팅 모의고사의 풀이방법에서 사용된 것 처럼 <일정 길이를="" 갖는="" 패턴="">을 인풋값의 총 길이로 나누어서 알아내는 방법(5월23일의 요일을 구해라? 2016년 1월1일이 금요일부터 시작이니까 1월(31일)+2월(29일)...+5월(31일)을 모두 더한 후 일주일 주기의 패턴인 7로 나눈 나머지를 적용.)

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]