Lv.1 2016년

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

코딩테스트 연습 - 2016년

기본틀

class Solution {
    public String solution(int a, int b) {
        String answer = "";
        return answer;
    }
}

첫번째 시도

class Solution {
    public String solution(int a, int b) {
        
        int n = 0;
        
        switch(a){
            case 1:
                n = 0;
                break;
            case 2:
                n = 31;
                break;
            case 3: 
                n = 31+29;
                break;
            case 4:
                n = 31+29+31;
                break;
            case 5: 
                n = 31+29+31+30;
                break;
            case 6:
                n = 31+29+31+30+31;
                break;
            case 7:
                n = 31+29+31+30+31+30;
                break;
            case 8:
                n = 31+29+31+30+31+30+31;
                break;
            case 9:
                n = 31+29+31+30+31+30+31+31;
                break;
            case 10:
                n = 31+29+31+30+31+30+31+31+30;
                break;
            case 11:
                n = 31+29+31+30+31+30+31+31+30+31;
                break;
            case 12:
                n = 31+29+31+30+31+30+31+31+30+31+30;
                break;
        }
        
        n += b;
        String[] arr = {"THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED"};
        int r = n % 7;
        String answer = "";
        answer += arr[r];
        
        return answer;
    }
}

음 주의할 점 : swith ~ case 쓸 때 break 해줘야 계속 이어지지 않는다!

성공!

했지만 당연히 마음에 안든다!

그래서 다른 사람의 풀이를 확인했는데, 아 왜 이렇게 풀 생각이 안들었지? ↓

두번째 시도

class Solution {
    public String solution(int a, int b) {
        
        int n = 0;
        
        int[] arrInt = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
        
        for(int i = 0; i < a-1; i++){
            n += arrInt[i];
        }
        
        n += b;
        String[] arr = {"THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED"};
        
        int r = n % 7;
        String answer = "";
        
        answer += arr[r];
        
        return answer;
    }
}

이렇게 풀면 된다~!