알고리즘/프로그래머스

[211223][210104] 2016년**(/)

감자탈출기 2021. 12. 23. 13:28

문제 설명

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

입출력 예

abresult
5 24 "TUE"

 

윤년 : 4년에 한번 2월이 29일까지 있는 해

class Solution {
    public String solution(int a, int b) {
        String answer = "";
        //인덱스 1에 1/1 요일 맞춤
        String[] day = {"THU", "FRI","SAT","SUN","MON","TUE","WED"};
        //인덱스 1에 1월부터 월별 날짜 맞춤
        int[] month = {0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
        int all = 0;
        //해당월 날짜인 b 더하기 이전 날짜들 더함 ex) 3월이면 1월 + 2월 dates 
        for(int i = 1; i < a; i++){
            all += month[i];
        }
        //인덱스 1에 요일 맞췄기([1]일 금요일) 때문에 바로 7로 나누면 됨
        answer = day[(all + b)%7];
        //만일 인덱스 0으로 맞췄을 경우 d-1 필요. answer = day[(all + b - 1)%7];
        return answer; 
    }
}

 

 

다른 방식으로 복습

class Solution {
    public String solution(int a, int b) {
        String answer = "";
        String[] day = {"SUN","MON","TUE","WED","THU", "FRI","SAT"};
        int[] month = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
        String[] all = new String[366];
        int dates = 0;
        for(int i = 0; i < all.length; i++){
            all[i] = day[(i + 5) % 7]; //나머지를 구하는 방식으로 배열범위 넘지 않도록 함
        } //all 안에 0~365 요일 채워짐, 1월 1일 = 인덱스 0 -> "FRI" 
        for(int i = 0; i < a - 1; i++){
            dates += month[i];
        }
        answer = all[(dates + b - 1) % 7]; // 1월 1일도 하루 지난 걸로 되기 때문에 인덱스 맞추려면 -1 해주어야 한다. 1월1일 FRI -> all[0] 이므로  
        return answer; 
    }
}

[220104] 위에 공부했던 방식보다 조금 간단하게 했음

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

 
리뷰 : dates += b-1;  로 애초에 날짜 계산할 때 빼주지 않고

String answer = date[dates%7-1]; <- 여기에서 빼주면 테스트 케이스 몇 건에서는 런타임 에러 발생

이유 : 날짜(dates)가 7일짜리 case면 index 값이 -1이 나오게 되므로 에러.