[Swift Algorithm] 2016년 Programmers(Lv.1)

문제 설명

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"

 

내 문제 풀이

func solution(_ a:Int, _ b:Int) -> String {
    let days = ["FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"]
    let months = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    var dayCount = 0 
    
    for i in 0..<a-1 {
        dayCount += months[i]
    }
    
    return days[(dayCount + b - 1) % 7]
}

• 1월 1일이 금요일로 주어졌으므로, 금요일부터 요일 배열을 생성해주었다.
• 달 별 날짜를 저정한 배열도 따로 만들어주었다.
• 입력된 월의 전 달까지의 날짜 수를 누적해주었고, 리턴문에서 해당 월의 요일을 구해 반환해주었다.

 

인상적인 풀이

func solution(_ a:Int, _ b:Int) -> String {
    let w = ["THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED"]
    let monthDay = [ 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    let totalDay = monthDay[0..<a-1].reduce(0, +) + b

    return w[totalDay % 7]
}

• 내 풀이의 for문을 reduce를 사용하여 한 줄로 줄여주었다.

 

💡 피드백

• 배열 안의 수를 더할 때는 reduce를 사용하면 간편하다,!
• 윤년 문제를 여기저기서 많이 발견할 수 있었다,, 익숙하게 풀 수 있도록 많이 연습하면 좋을 것 같다.

 

---

 

🖍 [ 추가 ] 1주일 후 다시 풀어보기

func solution(_ a:Int, _ b:Int) -> String {
    let days = ["FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"]
    let months = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    return days[(months[0..<(a-1)].reduce(0, +) + b - 1) % 7]
}

• reducea를 잘 활용하여 훨씬 깔끔하게 풀었다,!!

 

---

 

문제

https://programmers.co.kr/learn/courses/30/lessons/12901