[JAVA/알고리즘] 프로그래머스 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일 같은 날짜는 주어지지 않습니다. )

입출력 예

---

a	b	result
5	24	"TUE"

나의 풀이

---

class Solution {
  public String solution(int a, int b) {
        int totalDay = 0;
        String[] DayOfWeek = {"SUN","MON","TUE","WED","THU","FRI","SAT"};
        for (int i = 1; i < a; i++) {
            if (i == 4 || i == 6 || i == 9 || i == 11) {
                totalDay += 30;
            } else if (i == 2) {
                totalDay += 29;
            } else {
                totalDay += 31;
            }
        }
        totalDay += b+4;
      return DayOfWeek[totalDay%7];
  }
}

 

리턴용 문자열 배열 DayOfWeek를 생성해주고, 넘어온 월 만큼 반복문을 동작해 해당년 1월1일부터 a월까지의 날짜 수를 totalDay에 더해줍니다. 그 후 a월의 해당일인 b만큼 totalDay에 값을 더해주고 2016년 1월1일이 금요일이였기 때문에 +4를 해준다음에 %연산을 통해 해당 요일을 추출 DayOfWeek의 해당 인덱스를 리턴해줍니다.

 

 

마치며 

 

넘어오는 날짜 자체에서 유효한 데이터 값을 넘겨주었기 때문에 어렵지 않게 풀 수 있던 알고리즘이였습니다.

 

다른 사람의 풀이

class TryHelloWorld
{
    public String getDayName(int month, int day)
    {
        Calendar cal = new Calendar.Builder().setCalendarType("iso8601")
                        .setDate(2016, month - 1, day).build();
        return cal.getDisplayName(Calendar.DAY_OF_WEEK, Calendar.SHORT, new Locale("ko-KR")).toUpperCase();
    }
    public static void main(String[] args)
    {
        TryHelloWorld test = new TryHelloWorld();
        int a=5, b=24;
        System.out.println(test.getDayName(a,b));
    }
}