https://programmers.co.kr/learn/courses/30/lessons/12901
코딩테스트 연습 - 2016년
2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까
programmers.co.kr
풀이 과정
1월 1일과 a월 b일이 몇 일 차이가 나는지 계산 후 나머지 연산을 통해 어떤 요일인지 출력해주면 된다. 각 월의 일수와 요일을 리스트에 저장하여 문제를 해결하였다.
소스 코드
def solution(a, b):
wol = [0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
week = ["FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"]
answer = b - 1
subwol = a - 1
for i in range(subwol + 1):
answer += wol[i]
answer %= 7
answer = week[answer]
return answer'알고리즘 문제 풀이 > 프로그래머스' 카테고리의 다른 글
| 프로그래머스 - 나머지가 1이 되는 수 찾기 [파이썬] (0) | 2022.06.15 |
|---|---|
| 프로그래머스 - 최소직사각형 [파이썬] (0) | 2022.06.15 |
| 프로그래머스 - 두 개 뽑아서 더하기 [파이썬] (0) | 2022.05.26 |
| 프로그래머스 - 예산 [파이썬] (0) | 2022.05.26 |
| 프로그래머스 - 3진법 뒤집기 [파이썬] (0) | 2022.05.11 |