[Java/프로그래머스] 2016년

https://school.programmers.co.kr/learn/courses/30/lessons/12901

프로그래머스

 

 

1. 문제설명

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일같은 날짜는 주어지지 않습니다)
입출력 예
a b result
5 24 "TUE"

 

2. 문제풀이

class Solution {
    public String solution(int a, int b) {
        String answer = "";
        int monthdays31 = 31;
        int monthdays30 = 30;
        int monthdays29 = 29;
        
        int sum = 0;
        
       
        switch(a) {
            case 2 :
                sum = monthdays31;
                break;
            case 3 :
                sum = monthdays31 + monthdays29;
                break;
            case 4 :
                sum = monthdays31*2 + monthdays29;
                break;
            case 5 :
                sum = monthdays31*2 + monthdays29 + monthdays30;
                break;
            case 6 :
                sum = monthdays31*3 + monthdays29 + monthdays30;
                break;
             case 7 :
                sum = monthdays31*3 + monthdays29 + monthdays30*2;
                break;
            case 8 :
                sum = monthdays31*4 + monthdays29 + monthdays30*2;
                break;
            case 9 :
                sum = monthdays31*4 + monthdays29 + monthdays30*3;
                break;
            case 10 :
                sum = monthdays31*5 + monthdays29 + monthdays30*3;
                break;
             case 11 :
                sum = monthdays31*5 + monthdays29 + monthdays30*4;
                break;
            case 12 :
                sum = monthdays31*6 + monthdays29 + monthdays30*4;
                break;
        };

      
        
        switch((sum + b)%7) {
            case 1 :
            answer = "FRI";
             break;
            
            case 2 :
            answer = "SAT";
             break;
            
            case 3 :
            answer = "SUN";
             break;
            
            case 4 :
            answer = "MON";
             break;
            
             case 5 :
            answer = "TUE";
             break;
            
            case 6 :
            answer = "WED";
             break;
            
            case 0 :
            answer = "THU";
             break;
        }
        
        return answer;
    }
}

class Solution {
    public String solution(int a, int b) {
        String[] daysOfWeek = {"THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED"};
        int[] daysInMonth = {0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
        
        int totalDays = 0;
        
        for (int i = 0; i < a; i++) {
            totalDays += daysInMonth[i];
        }
        totalDays += b;
        
        return daysOfWeek[totalDays % 7];
    }
}

 

하드코딩으로 풀었던 솔루션을 switch문을 버리고 배열 함수를 지정해서 for문으로 돌려서 누적합을돌려서 배열지정으로 해주었다.