[ 코딩 테스트 예제 ] 07. 2016( 특정 날짜 요일 구하기 ) - Java

문제 

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

---

풀이

class Solution {
    public String solution(int a, int b) {
        String answer = "";
        String[] date = {"FRI","SAT","SUN","MON","TUE","WED","THU"};
        int[] date_per_month = {0,31,29,31,30,31,30,31,31,30,31,30,31};
        int month_stack = 0;
        for(int i = 1; i<a; i++){
            month_stack += date_per_month[i];
        }
        month_stack += b;
        while(month_stack > date.length ){
            month_stack-=date.length;
        }
        answer = date[month_stack-1];
        return answer;
    }
}

 

Java의  Date, Calendar 클래스를 사용하면 쉽게 해결할 수 있겠지만

그래도 나름 코딩 테스트 이기때문에! 생짜로 풀어보았다.

 

2016년의 매 월별 날짜를 기준으로 넘겨받은 달까지 month_stack에 더한다.

추가로 해당 월의 날짜까지 더해준 후에  date.length 밑으로 내려올 때까지 -= date.length를 한다.

( 달력 빨리 넘기기 같은 느낌 )

 

 

마지막으로 moth_stack -1을 하는 이유는 1월 0일은 없기 때문.

위 코드를 수행하면 특정 날짜의 요일을 구할 수 있다. (2016년 한정)

 

 

 

 

문제 출처

https://school.programmers.co.kr/learn/challenges?order=recent&languages=java 

코딩테스트 연습 | 프로그래머스 스쿨