코딩테스트/프로그래머스 level 1

[프로그래머스 level_1] 2016년 for JAVA

냠냠:) 2020. 3. 26. 02:17

[문제 설명]

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

[풀이]

이전 달들의 날짜와 현재 달의 날을 더한거에서 요일을 구하면 되는 문제였습니다. 그리고 1월 1일을 금요일인 것을 참고해 몇요일인지 구하면 되겠습니다.

 

[코드]

1
2
3
4
5
6
7
8
9
10
11
12
public static String solution(int a, int b) {
        int[] months = {0,31,29,31,30,31,30,31,31,30,31,30,31};
        String[] days = {"THU","FRI""SAT","SUN","MON","TUE","WED"}; 
        int sum = 0;
        for(int i = 0; i < a; i++) {
            sum += months[i];
        }
        sum += b;
        
        sum = (sum) % days.length;
        return days[sum];
    }
cs

느낀점 : 모듈러 연산을 위해 7일 째인 "THU"을 인덱스 0에 두는 것이 인상 깊었습니다.

 

반응형