알고리즘/프로그래머스
[프로그래머스] 2016 (python)
김홍중
2021. 5. 26. 22:51
def solution(a, b):
day = ['SUN', 'MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT']
all_day = 366
year = []
month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
target_day = 0
for i in range(366):
year.append(day[(i + 5) % 7])
for i in range(a-1):
target_day += month[i]
target_day += b - 1
return year[target_day]