[test] 2016년

src: 프로그래머스

문제 설명
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일같은 날짜는 주어지지 않습니다)

python

내 풀이

def solution(a, b):
    days = ["SUN","MON","TUE","WED","THU","FRI","SAT"]
    arr = [4,5,6,0,1,2,3]
    months = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]

    md = 0
    for i in range(a-1):
        md += months[i]
    idx = (md + b)%7
    idx2 = arr[idx]
    
    return days[idx2]

days로 요일 배열주기
1월 1일과 비교해서 몇일 더 더해졌나 구하고 더해진 날짜수를 7로 나눴을 때 나머지에 따라 요일이 달라짐.
arr는 그 나머지에 따른 days의 인덱스.
months로 월 별 날짜 수 미리 적어주기.

md 는 month says로 a월 b일에서 a월 이전 달들의 일수를 더해줌.
md와 b를 더한 총 일에서 7로 나눈 나머지를 idx 변수에 저장.
arr 에서 idx 자리에 있는 숫자가 days의 인덱스가 됨.
days 의 idx2에 해당하는 요일 리턴

다른풀이

def getDayName(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]

기본 논리는 비슷한데, 아얘 days배열을 FRI부터 시작해서 한 단계 축소시킴.
sum함수 사용해서 md구함.
한꺼번에 한줄로 써서 보기 좋음.