[프로그래머스] Lv.1 2016년 java

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

코딩테스트 연습 - 2016년

 

문제

2016년 1월 1일은 금요일. 2016년 a월 b일은 무슨 요일일까요? 두 수 a,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수를 완성하시오. 요일의 이름은 일요일부터 토요일까지 각각 SUN, MON, WED, THU, TRI, SAT. 예를 들어, A=5, B=24라면 5월 24일은 화요일이므로 문자열 "TUE"를 반환.

제한조건

2016년은 윤년

2016년 a월 b일은 실제로 있는 날(13월 26일이나 2월 45일같은 날짜는 주어지지 않음)

 

.java

class Solution {
    public String solution(int a, int b) {
        String answer = "";
        //1월1일이 금요일이니까 계산하기 편하도록 일주일 중 금요일이 시작처럼 배열
        String[] day={"FRI","SAT","SUN","MON","TUE","WED","THU"};
        int[] last={31,29,31,30,31,30,31,31,30,31,30,31};
        //2016년의 몇번째 날인지 저장할 변수
        int date=0;
        
        //1.월별 마지막 날 수의 합
        for(int i=0; i<a-1; i++){
            date += last[i];
        }
        //2.a-1달까지의 일수 + a달의 일수
        date += b-1;
        
        //3.일수%7로 요일 구해서 리턴
        return day[date%7];
    }
}

 

실행 결과