Programmers

2016년

h32j00 2020. 12. 29. 22:25

문제

문제

처음 작성한 코드

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

 

코드 리뷰 후

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

 

깨달은 점

list[:n] 의 사용법을 상기시켰다.

파이썬 어렵다..