[프로그래머스] Lv.1 2016년

문제 : Lv.1 2016년 https://school.programmers.co.kr/learn/courses/30/lessons/12901 언어 : JAVA 체감 Level : ★ ☆ ☆ Review: 달력 굴러가는 로직 이해

 

💡

일단 문제에서 윤년이라고 주어졌으므로 2월은 29일이 된다. (윤년아니면 28일)

나머지는 1~7은 짝수월이 30일을 가리키고

8~12부터는 홀수월이 30일을 가리킨다.

이점을 생각하고 주어진 월에 맞게 일 수를 누적해간다.

 

해당 월에 다다르면 누적을 멈추고

`sum + b` 를 해준다.

 

누적된 값은 7로 나누면 요일을 알아낼 수 있다. (0~6)

이때 나는 배열로 쉽게 반환시키도록 구현했다.

1월 1일이 `FRI` 였으므로 `FRI` 부터 진행

 

📍 이때 런타임 오류가 떴는데 0 일때 처리를 해줘야하는 걸 놓쳤다

그래서 마지막엔 THU 가 -1 이 되어버려

런타임 오류가 발생하는 것을 막기위해 조건문을 설정해서 개별 처리해주었다.

 

class Solution {
    public String solution(int a, int b) {
        String answer = "";
        String[] cal = {"FRI","SAT","SUN","MON","TUE","WED","THU"};
        int sum = 0;
        int mon = 1;
        
        // 31 or 30 or 윤년 : 29
        while(mon < a){
            
            if (mon == 2){
                sum += 29;
            }
            else if(mon >= 1 && mon <= 7){
                if(mon % 2 == 0 ){
                    sum += 30;
                }else {
                    sum += 31;
                }
            }else if( mon <= 12){
                if(mon % 2 == 0){
                         sum += 31;
                }else {
                    sum += 30;
                }
            }
            
        mon ++;
        }
      
        int result = (sum+b)%7-1;
        
        if (result < 0){ // 나누어떨어지는경우에는 -1 이 되어버림 개별처리
            answer = cal[6];
        }else {
             answer = cal[result];
        }
        
        
        return answer;
    }
}