8. 2016년 - 프로그래머스 코딩테스트

문제 설명

• 2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까지 각각 SUN,MON,TUE,WED,THU,FRI,SAT입니다. 예를 들어 a=5, b=24라면 5월 24일은 화요일이므로 문자열 "TUE"를 반환하세요.

제한 조건

• 2016년은 윤년입니다.
• 2016년 a월 b일은 실제로 있는 날입니다. (13월 26일이나 2월 45일같은 날짜는 주어지지 않습니다)

# 입출력 예
a = 5
b = 5
# 'TUE'

# python의 날짜 라이브러리를 불러온다
from datetime import datetime, date

def solution(a, b):
    # 월~일요일 까지를 days에 리스트로 할당한다.
    days = ['MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT', 'SUN']

    # date의 weekday함수를 활용하여 2016년도의 a월 b일의 요일을 구한다.
    # 이때 요일이 0~6까지 숫자로 나오기에
    # days의 인덱싱으로 해당 요일을 구한다.
    return days[date.weekday(date(2016,a,b))]

solution(a, b)

# 'THU'

다른 사람의 풀이

# 입출력 예
a = 5
b = 5
# 'TUE'

def getDayName(a,b):
    answer = ""
    if a>=2:
        b+=31
        if a>=3:
            b+=29#2월
            if a>=4:
                b+=31#3월
                if a>=5:
                    b+=30#4월 # a가 5 -> 즉 5월 이기에 b + (31+29+31+30)이 실행된다.
                    if a>=6:
                        b+=31#5월
                        if a>=7:
                            b+=30#6월
                            if a>=8:
                                b+=31#7월
                                if a>=9:
                                    b+=31#8월
                                    if a>=10:
                                        b+=30#9월
                                        if a>=11:
                                            b+=31#10월
                                            if a==12:
                                                b+=30#11월
    # 현제 a = 5
    # b = 126 이다
    # b를 일주일인 7로 나눈 후 나머지를 계산한다.
    b=b%7
    # 126 % 7 = 0 이다.

    # 2016년 1월 1일이 금요일이기에
    # b==1일때 금요일부터 시작한다.
    if b==1:answer="FRI"
    elif b==2:answer="SAT" 
    elif b==3:answer="SUN"
    elif b==4:answer="MON"
    elif b==5:answer="TUE"
    elif b==6:answer="WED"
    else:answer="THU"
    return answer


print(getDayName(5,24))

# 'THU'