Programmers Lv.1 - 2016년[Java]

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

나의 풀이

class Solution {
    public String solution(int a, int b) {
        int cnt = 0;
        String[] days = {"FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"};

        for (int i = 1; i < a; i++) {
            if (i ==1 || i == 3 || i == 5 || i == 7 || i == 8 || i == 10 || i == 12)
                cnt += 31;
            else if (i==2)
                cnt += 29;
            else
                cnt += 30;
        }

        cnt += b;
        cnt--;

        return days[cnt%7];
    }
}

수정한 버전

class Solution {
    public String solution(int a, int b) {
        int cnt = 0;
        String[] days = {"FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"};
        int[] date = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};

        for (int i = 1; i < a; i++) {
            cnt += date[i-1];
        }

        cnt += b;
        cnt--;

        return days[cnt%7];
    }
}