알고리즘/프로그래머스
[Python] 2016년
컴공맨
2021. 2. 21. 20:05
코딩테스트 연습 - 2016년
2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까
programmers.co.kr
풀이
총 일수를 구하고 일주일은 7일 이므로 총 일수에 7을 나머지연산하여 해결했습니다.
코드
def solution(a, b):
answer = ''
month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
day = ["THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED"]
total_day = b
for i in range(0, a - 1):
total_day += month[i]
answer = day[total_day % 7]
return answerpyo7410/Algorithm
1일 1커밋을 목표로! Contribute to pyo7410/Algorithm development by creating an account on GitHub.
github.com