[프로그래머스_Kotlin] 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일같은 날짜는 주어지지 않습니다)

 

입출력 예

 

내 생각

예전에 들었던 둠스데이로 풀어야하나 하다가 둠스데이 알고리즘을 찾아보려다가 큰 날짜가 아닌 16년도만 찾으면되니까

- 1.1일은 금요일이다. 예시를 들어 풀어보면 1.5일은 토->일->월->화

- ex1) 1.20일은 20일 - 1일 = 19 

    + 19 % 7 = 5

    + (금요일 + 5) = 수요일

- ex) 3.5일은 3.5일 - 1.1일

    + 31일(1월) + 29일(2월) + 5(3.5일의 5일) -1일(1.1일의 1일) = 64

    + 64 % 7 = 1

    + (금요일+1) = 토요일

 

1) 달마다 몇일인지 나열(윤년이라 2월은 29일), 요일 나열

2)  a달만큼 앞에 달 더하고 요일 더한 뒤 % 7(요일) 한뒤 나머지를 요일 index 출력

(설명에 소질이 없는 것 같다,,,)

 

풀이

class Solution {
    fun solution(a: Int, b: Int): String {
        val monthList = listOf(31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31)
        val dayList = listOf("THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED")

        var day = ((0 until a - 1).map { monthList[it] }.sum() + b) % 7

        return dayList[day]
    }
}

 

 

다른 사람의 풀이

class Solution {
    fun solution(a: Int, b: Int): String {
        val week = listOf("THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED")

        var answer = ""
        var days = b

        for (i in 1 until a) {
            days += when (i) {
                1, 3, 5, 7, 8, 10, 12 -> 31
                2 -> 29
                else -> 30
            }
        }

        answer = week[days % 7]
        return answer
    }
}