[프로그래머스/Programmers] 2016년 - Python

Link

코딩테스트 연습 - 2016년

 

Solution

def solution(a, b):
    weekday = ['SUN', 'MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT']
    month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]

    count = b - 1

    for m in range(a - 1):
        count += month[m]

    return weekday[count % 7 - 2] # 2016년 1월 1일: 금요일 -> -2

🔑 key point: Indexing

 

Other Solution

import datetime

def solution(a,b):
    weekday = ['SUN', 'MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT']

    return weekday[datetime.datetime(2016, a, b).weekday() + 1]