[python]programmers-2016년

level1

문제 : 2016년

 

 

1. 내풀이

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

1월 1일이 금요일이기때문에 7로 나눴을 때의 나머지가 1이면 금요일이다.

따라서 day라는 리스트를 지정했는데 요일의 인덱스를 나머지값과 동일하도록 설정했다.

 

1월1일부터 현재까지 며칠이 지났는지 알아야 하기때문에 month라는 리스트에 각 월의 날짜 값을 저장해두었다.

 

현재까지 지난 일수를 7로 나눠서 나온 나머지를 인덱스로 해서 day에서 요일을 꺼내면 된다.