[프로그래머스 파이썬] 2016년

일단 해볼게·2023년 11월 8일
0

프로그래머스

목록 보기
77/106

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

def solution(a, b):
    day = ["THU","FRI","SAT","SUN","MON","TUE","WED"] #1월1일이 금요일이므로 7로 나눴을 때 인덱스 1이 금요일로 나오게 배열을 설정
    mon = [31, 29, 31, 30, 31, 30 , 31, 31, 30, 31, 30, 31] # 각 달의 날짜만큼 배열로 저장, 2월은 윤년이라 29일
    
    # mon배열의 (a-1)월까지 더한수에 <- 3월이면 2월까지의 날짜 다 더한수
    # 구하고자하는 b날짜를 더하고 7로 나눈 나머지의 인덱스에 해당하는 수 반환
    return day[(sum(mon[:a - 1]) + b) % 7] 
profile
시도하고 More Do하는 백엔드 개발자입니다.

0개의 댓글