2016년

1. 문제

문제 설명

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	b	result
5	24	TUE

 

2. 풀이 과정

import datetime

def solution(a, b):
    week = ['FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED', 'THU']

    first_day = datetime.datetime(2016, 1, 1)
    target = datetime.datetime(2016, a, b)

    diff = target - first_day

    answer = week[(diff.days % 7)]

    return answer

datetime 이라는 모듈을 알고 있었으나 저번 문자열을 정수로 만들기 문제에서 int() 로 한 번에 풀고 싶지 않아

 

나름 datetime 모듈을 덜 활용해보려고 했으나...그냥 쓸 걸 그랬다.

 

 

차라리 다음과 같이

def solution(a, b):
    week = ['MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT', 'SUN']

    target = datetime.datetime(2016, a, b).weekday()

    return week[target]

요일을 알려주는 weekday() 까지 같이 쓰던가

 

 

아니면

def solution(a, b):
    months = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    week = ['FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED', 'THU']

    answer = week[(sum(months[:(a - 1)]) + b - 1) % 7]

    return answer

아예 datetime 모듈을 쓰지 않고 풀었으면 더 좋았을 거라 생각한다.

 

3. 다른 사람의 풀이

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월
                    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월
    b=b%7

    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

전설의 아트 코딩...ㅋㅋㅋㅋㅋㅋㅋㅋ

 

밑에 성지 순례 댓이 많길래 나도 올해 잘 됐으면 하는 바람으로 다른 사람 풀이에 넣어본다ㅎㅎ