<문제>
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"를 반환하세요.
<풀이>
def solution(a, b):
m = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
d = ['THU','FRI','SAT','SUN','MON','TUE','WED']
dday = sum(m[:a-1]) + b
return d[dday%7]
<다른 풀이>
방법 1.
def getDayName(a,b):
answer = ""
if a>=2:
b+=31
if a>=3:
b+=29
if a>=4:
b+=31
if a>=5:
b+=30
if a>=6:
b+=31
if a>=7:
b+=30
if a>=8:
b+=31
if a>=9:
b+=31
if a>=10:
b+=30
if a>=11:
b+=31
if a==12:
b+=30
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
방법 2.
import datetime
def getDayName(a,b):
t = 'MON TUE WED THU FRI SAT SUN'.split()
return t[datetime.datetime(2016, a, b).weekday()]
방법 3.
def getDayName(a,b):
day_name = ['THU', 'FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED']
month_dict = {
"1":31,
"2":29,
"3":31,
"4":30,
"5":31,
"6":30,
"7":31,
"8":31,
"9":30,
"10":31,
"11":30,
"12":31
}
days = 0
for i in range(1, a):
days += month_dict[str(i)]
days += b
index = days % 7
return day_name[index]
방법 4.
def solution(a, b):
days = ["SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT"]
dayLen = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
now = 5
for i in range(0, a - 1) :
now += dayLen[i]
answer = (now + b - 1) % 7
return days[answer]
<느낀점>
사실 방법1 경우를 가장 먼저 생각했다. 도저히 어떻게 코드를 구현해야할 지 감이 안와서 노가다로 하다가 결국 list형식으로 구현했다. 끝까지 코드를 완성시키는 것 또한 너무나 본받을 점이다.