[프로그래머스] 2016년 (파이썬)

 

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

코딩테스트 연습 - 2016년

 

 

---

 

 

<나의 풀이>

def solution(a, b):
    answer = ''
    day = ['FRI','SAT','SUN','MON','TUE','WED','THU']
    month = [31,29,31,30,31,30,31,31,30,31,30,31]
    
    answer = day[(sum(month[:a-1])+b) % 7 - 1]
    return answer

2016년이 금요일부터 시작이라고 해서 day를 'FRI'로 시작했다.

month는 달 별로 일 수를 넣어주었다. 

1월의 일 수부터 특정 날짜까지의 일 수를 모두 더해준 후,

요일 수인 7로 나눠주었다.

마지막으로 인덱스는 0부터 시작하니까 나머지 값에서 1을 빼주면

특정 날짜의 요일을 구할 수 있다.

문제에 나온 예시를 기준으로 식으로 써보면,

5월 24일의 요일은

=> ( 1월(31) + 2월(29) + 3월(31) + 4월(30) + 5월(24) ) % 7 - 1 = 4

=> day[4]는 'TUE'

 

다른 사람들의 코드를 보니까 다들 예쁜 모양으로 코드를 짰다🤣🤣

그런 열정에 절로 박수가 나왔다👍🏻