본문 바로가기
알고리즘/프로그래머스

[Python] 2016년

by 컴공맨 2021. 2. 21.
 

코딩테스트 연습 - 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 answer

 

pyo7410/Algorithm

1일 1커밋을 목표로! Contribute to pyo7410/Algorithm development by creating an account on GitHub.

github.com

 

'알고리즘 > 프로그래머스' 카테고리의 다른 글

[Python] 비밀지도  (0) 2021.02.28
[Python] 키패드 누르기  (0) 2021.02.27
[Python] K번째수  (0) 2021.02.20
[Python] 체육복  (0) 2021.02.13
[Python] 모의고사  (0) 2021.02.12