[프로그래머스/Java] 2016년

 

출처

프로그래머스 2016년 
https://programmers.co.kr/learn/courses/30/lessons/12901

문제

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

 

코드 

 

class Solution {
  public String solution(int a, int b) {
      String answer = "";      
      String weeks [] = {"SUN","MON","TUE","WED","THU","FRI","SAT"};
      int months [] = {0,31,29,31,30,31,30,31,31,30,31,30,31};
      
      int month;
      int sum = 0;
      
      for (int i=0; i<a; i++){
          sum += months[i];
      }      
      
      int date = sum + b; 
    
      switch (date%7){
          case(0) : answer = weeks[4]; break;
          case(1) : answer = weeks[5]; break;
          case(2) : answer = weeks[6]; break;
          case(3) : answer = weeks[0]; break;
          case(4) : answer = weeks[1]; break;
          case(5) : answer = weeks[2]; break;
          case(6) : answer = weeks[3]; break;
      }
          
      return answer;
  }
}

설명

- 2016년은 윤달이므로 2월은 29일까지 있다. 
그에 맞는 months 배열을 만들어 준다. 
- for를 이용하여 몇월인지 확인하여 해당 달의 전달까지의 날짜를 더해준다. 
- 더한 값에 현재 날짜를 더해서 date 변수를 만들어 준다. 
- switch를 이용하여 (모두 더한 날짜)나누기 7 하여 나머지를 구해준다. 
   나머지 값에 맞는 케이스를 작성한다. 
- 예를들어 1월 1일은 모든 날짜가 1이고, 7로 나눈경우 나머지는 1이다. 
case (1) 은 weeks 에서 FRI를 출력하도록 해주고, break를 넣어준다.