[알고리즘]기초2-2016년

2016년

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

def solution(a, b):
    answer = ''
    days=['FRI','SAT','SUN','MON','TUE','WED','THU']
    monthes={1:31,2:29, 3:31, 4:30, 5:31, 6:30, 7:31, 8:31, 9:30, 10:31, 11:30, 12:31}
    diff_days=0
    
    if a==1:
        answer=days[(b-1)%7]
    else:
        for i in range(1,a):
            diff_days+=monthes[i]
        diff_days+=b
        answer=days[(diff_days-1)%7]
    return answer

👉🏻요일을 순서대로 넣은 days 리스트와 1월부터 12월까지 각 월의 일수를 저장한 딕셔너리 활용

• days 리스트와 monthes 딕셔너리 생성
• a가 1월일 때는 b-1을 7로 나눈 나머지를 인덱스로 하는 요일을 answer로 반환
  -a가 1월이 아닐 때는 for문을 이용하여 1월부터 a-1월 까지의 일 수를 모두 더하고 b를 더하여 diff_days를구함
• diff days-1을 7로 나눈 나머지를 인덱스로 하는 요일을 answer로 반환