알고리즘/프로그래머스
[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일같은 날짜는 주어지지 않습니다)
입출력 예
abresult5 | 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이 나오게 되므로 에러.