자기개발👨‍💻/코딩 알고리즘

[python] 2016년 (요일구하기)

천숭이 2021. 1. 3. 08:12
def solution(a, b):
    answer = ''
    month=[31,29,31,30,31,30,31,31,30,31,30,31] 
    result = {4:"MON",5:"TUE",6:"WED",0:"THU",1:"FRI",2:"SAT",3:"SUN"}
    answer = result.get(int(sum(month[:a-1])+b)%7)
    return answer

if문 사용했을때는 시간초과 발생