본문 바로가기
Preparing Coding Test/Programmers L1

[Java] 2016년

by weero 2020. 7. 14.

문제

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

 

코딩테스트 연습 - 2016년

2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까��

programmers.co.kr

 

 

코드

class Solution {
  public String solution(int a, int b) {
      String answer = "";
      String[] days = {"THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED"};
      int sum = 0;
      
      for(int i=1; i<a;i++){
          if(i<8){
              if(i==2) 
                  sum += 29;
              else if(i%2==0)
                  sum += 30;
              else
                  sum+= 31;
          }else {
              if(i%2==0)
                  sum+=31;
              else
                  sum+=30;
          }
      }
      sum+=b;
      
      answer=days[sum%7];
      
      return answer;
  }
}