[프로그래머스_1] 2016 JAVA

프로그래머스

 

 

CODE

class Solution {
    public String solution(int a, int b) {
        int[] month = {0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
        String[] week = {"FRI","SAT","SUN", "MON", "TUE", "WED", "THU"};
        
        int sum = 0;
        for(int i = 1; i < a; i++) {
            sum += month[i];
        }
        
        int idx = sum + b-1;

        return week[idx%7];
    }
}

풀이 포인트

• 1~12월까지 각 달의 일수를 저장한 배열 선언
• 요일을 저장한 배열 선언
  • 2016년 1월 1일은 금요일부터 시작이니 금요일부터 저장
• week의 idx가 0이어야 하는 경우(1월 1일 등)를 위한 b-1  계산

 

실행 결과