[1일1코테] 프로그래머스 2016년

점심시간에 시간이 남아서 간단하게 풀어보는 코딩문제

확실히 기업 문제가 아닌 이상 Level 1은 너무 쉬운거 같은데....

"윤년" 이거 초등학교? 중학교? 쯤에 날짜 계산하는 수학 문제에서 많이 나왔는데...
2월달이 28일까지 있는데, 29일이 되는 년도를 윤년이라고 하고 4년에 한번씩 발생? 하는걸로 기억한다.

* 풀이 방향 :

1. 요일 저장하는 리스트 생성
   2016년1월1일이 금요일이라고 하길래 금요일부터 저장
2. 윤년에 맞게 달 수 저장하는 리스트 생성
3. 달과 요일 가져와서 맞는 요일 찾기

• 최종 코드 :

def solution(a, b):
    answer = ''
    day = ['FRI','SAT','SUN','MON','TUE', 'WED','THU']
    
    #1월~12월 있는 달??
    month =[0, 31, 29, 31, 30,31,30,31,31,30,31,30,31]
    tmp = 0
    
    tmp = sum(month[:a]) + b
    
    answer = day[tmp%len(day)-1]
    
    return answer