문제
- 문제 설명
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일같은 날짜는 주어지지 않습니다)
- 입출력 예시
a b result 5 24 "TUE"
문제 풀이
문제가 생각보다 쉬워 보이면서도 어떻게 풀어가야 할지 고민을 했던 문제이다. 처음엔 달마다 switch case문을 이용해 풀어야 하나 라는 생각도 했지만 코드가 너무 지저분해 보일 것 같아 선택한 방법은 아니다.
그래서 만약 입력값이 5월 24일이면, (4월까지의 모든 일수 + 24일)을 하여 문제를 풀기로 생각했다.
아래 코드를 보자.
class Solution {
public String solution(int a, int b) {
String answer = "";
int[] days = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
String[] week = {"FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"};
int sum = 0;
for(int i = 0; i < a - 1; i++) sum += days[i];
sum += b - 1;
answer = week[sum % 7];
return answer;
}
}먼저, 2016년 1월 1일은 금요일이다. 금요일부터 시작하기 때문에 week 배열은 금요일부터 시작하도록 초기화했다. 그리고 days 배열은 월별 날짜의 수이다.
sum += b - 1을 하는 이유는 2016년 1월 1일은 금요일인데, 만약 입력값이 1월 1일이면, 배열에서는 0번째 인덱스를 불러와야 하기 때문이다. 그래서 1을 빼주도록 한다.
최종적으로 sum 값에서 7을 나누고 그 나머지 값이 요일이 되므로 그렇게 계산하면 된다.
Android Developer