[프로그래머스] Lv1. 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일같은 날짜는 주어지지 않습니다)

나의 풀이:

def solution(a, b):
    날짜수 = 1월 1일부터 a월 b일까지의 총 날짜수
				if 날짜수 % 7 == 0:
				    return "THU"
				elif 날짜수 % 7 == 1:
				    return "FRI"
				elif 날짜수 % 7 == 2:
				    return "SAT"
				elif 날짜수 % 7 == 3:
				    return "SUN"
				elif 날짜수 % 7 == 4:
				    return "MON"
				elif 날짜수 % 7 == 5:
				    return "TUE"
				elif 날짜수 % 7 == 6:
				    return "WED"

문제를 pseudo 코드로 풀어보면 위와 같을 것이다. 날짜수의 계산이 관건이다.

def solution(a, b):
		총 날짜수 =
		a가 1월이면 b 만으로 가능
		a가 2월이면 31 + b
		a가 3월이면 31 + 29 + b
		a가 4월이면 31 + 29 + 31 + b
		a가 5월이면 31 + 29 + 31 + 30 + b
		a가 6월이면 31 + 29 + 31 + 30 + 31 + b
		a가 7월이면 31 + 29 + 31 + 30 + 31 + 30 + b
		a가 8월이면 31 + 29 + 31 + 30 + 31 + 30 + 31 + b
		a가 9월이면 31 + 29 + 31 + 30 + 31 + 30 + 31 + 31 + b
		a가 10월이면 31 + 29 + 31 + 30 + 31 + 30 + 31 + 31 + 30 + b
		a가 11월이면 31 + 29 + 31 + 30 + 31 + 30 + 31 + 31 + 30 + 31 + b
		a가 12월이면 31 + 29 + 31 + 30 + 31 + 30 + 31 + 31 + 30 + 31 + 30 + b
		
		if 날짜수 % 7 == 0:
		    return "THU"
		elif 날짜수 % 7 == 1:
		    return "FRI"
		elif 날짜수 % 7 == 2:
		    return "SAT"
		elif 날짜수 % 7 == 3:
		    return "SUN"
		elif 날짜수 % 7 == 4:
		    return "MON"
		elif 날짜수 % 7 == 5:
		    return "TUE"
		elif 날짜수 % 7 == 6:
		    return "WED

위 과정을 잘 정리하면 되지 않을까?

def solution(a, b):
    month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    day = ['THU', 'FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED']
    num = sum(month[:a-1]) + b
    return day[num % 7]

달 별로의 요일수와 요일의 종류는 항상 반복되기 때문에 리스트로 계산하면 쉬울 것이라고 생각했다. 인덱스 슬라이싱을 통해 월별로 지나간 요일수의 합을 구하고, if문 대신 “날짜수 % 7”의 값을 곧바로 리스트 인덱스로 사용할 수 있도록 했다.

다른 사람들의 풀이:

def solution(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

노가다를 예술로 승화시켰다는 작품ㅋㅋㅋㅋㅋ

하지만 웃지말자, 나도 리스트 생각 못했으면 저리할 뻔 했으니…