[프로그래머스, Lv1] 2016년

 

코딩테스트 연습 - 2016년 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

 

문제설명

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

접근 방법

이 문제의 핵심 내용은 2016년도의 1~12월 각각 일 수를 구하는 것이었다.

매 달 일수가 30일로 통일되어 있는 것이 아닌,

1월부터 [ 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 ] 처럼 제각각인 만큼,

해당 내용을 해결해야 정확한 요일을 구할 수 있었기 때문이다.

 

처음에는 LocalDate.lengthOfMonth() 메서드를 사용하여 매 달의 일을 구할까 했지만, 총 12개의 숫자만 미리 갖고 있다면, LocalDate 객체 생성 / 재할당을 하지 않아도 된다는 생각이 들었다.

 

또한, 반환값( 요일 이름 )에 대해서도, if문을 사용하는 것이 아닌 배열에 넣어놓고 사용하기로 했다.

 

1차 코드

class Solution {
    public String solution(int a, int b) {
        int[] dayOfMonth = { 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
        String[] nameOfDay = { "SUN", "MON", "TUE" ,"WED", "THU", "FRI", "SAT"};
        int nowDay = 5;

        for(int i = 0; i < a - 1; ++i)
        {
            nowDay += dayOfMonth[i];
        }

        nowDay += b - 1;

        return nameOfDay[nowDay % 7];
    }
}

 

 

해당 코드로 작성하였고 통과 되었다.

[ 통과 ]