[JAVA] 프로그래머스 2016년

이번에 풀어볼 문제는 2016년이다.

 

문제 설명

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일 같은 날짜는 주어지지 않습니다.

 

그러면 이제 코드를 봐보자.

 

public String solution(int a, int b) {

        String[] day = {"FRI","SAT","SUN","MON","TUE","WED","THU"};
        int[] date = {31,29,31,30,31,30,31,31,30,31,30,31};
        int select = 0;

        for(int i = 0; i < a-1; i++){
            select += date[i];
        }
        select += b - 1;

        return day[select % 7];
    }

a는 월이고, b는 일이다.

String배열에 금요일부터 모든 요일을 다 넣어 놨다.(문제 설명에서 보다시피 첫날이 금요일 이기 때문)

다음 int 배열에는 달마다 마지막 날을 넣어 놨다.(1월은 31일, 2월은 윤년이니 29일....)

 

select는 몇 월 며칠을 숫자로 나타낸 값이다.

for문을 돌려 a-1까지 date 배열에 있는 값을 select에 더해준다.(7월이라면 6월까지는 다 더하는 것)

그다음 select에 b-1의 값을 더해준다.

 

b-1을 하는 이유는 1월 1일이 하루가 지난날이 아니기 때문이다.

만약 a, b에 1,1이 들어오고 b-1을 해주지 않는다면 day [select]는 day [1]이 되어 버리고 그러면 1월 1일은 금요일인데 SAT가 나오게 된다.

 

마지막으로 리턴할 때 select의 값을 요일 수 7로 나눈 나머지를 day에서 리턴 시켜 준다.