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

lv1-2016년 (python, 파이썬)

by bky373 2020. 10. 22.
""" 1번: 나의 풀이 """
def solution(a, b):
    days = [31,29,31,30,31,30,31,31,30,31,30,31]
    theday = ['FRI','SAT','SUN','MON','TUE','WED','THU']

    past_days = 0
    for x in range(a-1):
        past_days += days[x]
    past_days += b-1
    
    return theday[past_days%7]




""" 2번: 다른 사람의 풀이 """
def solution2(a, b):
    months = [31,29,31,30,31,30,31,31,30,31,30,31]
    days = ['FRI','SAT','SUN','MON','TUE','WED','THU']
    return days[(sum(months[:a-1])+b-1)%7]



- 확인하기
  - sum 함수를 활용하여 간단히 작성 가능 (알면서도 까먹는다..)

댓글