[프로그래머스] 2016년 문제풀이 python

🐒 문제

https://programmers.co.kr/learn/courses/30/lessons/12901#

✍ 나의 풀이

def solution(a, b):
    cal = [0,31,29,31,30,31,30,31,31,30,31,30,31]
    days = sum(cal[1:a])+b-1
    week = ["FRI","SAT","SUN","MON","TUE","WED","THU"]
    return week[days%7]

30일 31일 날짜세는 법을 헷갈렸다.

코드

def solution(a, b):
    cal = [0,31,29,31,30,31,30,31,31,30,31,30,31]
    	# 인덱스 i 월이 몇일까지 있는지 저장된 리스트
        
    days = sum(cal[1:a])+b-1
    	# a월 b일이 2016년 1월 1일로 부터 며칠 후인지 계산
        	# -1 하는 이유는 1월 1일 당일 제외하기위함
            
    week = ["FRI","SAT","SUN","MON","TUE","WED","THU"]
    	# 7일마다 반복되는 요일
    return week[days%7]