알고리즘 풀이/프로그래머스

[level1] 프로그래머스 - 2016년(JAVA)

데롱디롱 2021. 8. 29. 17:55
728x90

1월 1일이 'FRI'이므로 days[1]이 FRI가 되도록 요일정보를 담은 days배열을 만들었다.

String[] days = { "THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED" };

 

1~12월까지의 일 수를 months배열에 넣고

int[] months = { 0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };

입력되는 월(a)에 따라 1 ~ a 전 월까지의 일수의 합 + 일(b)를 해주었다.

 

그리고 이것을 7로 나눈 나머지가 바로 요일!!

 

 

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

        int n = b;
        for (int i = 1; i < a; i++)
            n += months[i];

        return days[n % 7];
    }
}
댓글수0