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

입출력 예

풀이

달력을 보면 달마다 1일이 시작하는 날이 다르기 때문에 요일이 매번 달라진다.

이번 문제는 년도가 2016년이라고 정해져있기 때문에, 매 월마다 시작하는 요일을 저장해뒀다.
(아마 년도가 정해져있지 않고 랜덤으로 주어졌다면 이렇게 풀지 않았을 것이다.)

만약, 2016년 6월이라면 1일까지의 요일이 월, 화, 수 총 세개이므로 3을 저장한다.
요일의 문자열을 저장한 배열 day = ["SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT"] 이 있을 때,
6월 n일의 요일은 day[(3+n-1)%7] 이다.

day[(3+n)%7-1] 로 하지 않은 이유는, index가 0 아래로 내려가면 안되기 때문이고 나머지연산이기 때문에 -1을 먼저 해줘도 상관없기 때문이다.

소스코드

function solution(a, b) {
    let day = ["SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT"];
    // 해당 달의 첫째 날까지의 공석
    let firstDay = [5, 1, 2, 5, 0, 3, 5, 1, 4, 6, 2, 4];
    
    let index = (firstDay[a-1]+b-1)%7;
    return day[index];
}