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를 반환하세요.
a | b | result |
---|---|---|
5 | 24 | “TUE” |
지금까지 해온 프로젝트에 경험으로 바로 생각난 알고리즘은
Apple Kit를 활용한 DateFormat과 Calendar를 이용하여 문제를 풀게 되었다.
import Foundation
let dayStrings: [String] = ["SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT"]
func solution(_ a:Int, _ b:Int) -> String {
let dateFormat = DateFormatter()
dateFormat.dateFormat = "yyyy-M-d"
let cal = Calendar.current
guard let date = dateFormat.date(from: "2016-\(a)-\(b)")else {
return ""
}
let components = cal.dateComponents([.weekday], from: date)
guard let day = components.weekday else {
return ""
}
return dayStrings[day - 1]
}
문제를 풀고 난 후에 다른 분들에 풀이를 봤는데, 좀 더 가볍게 생각해서 풀어도 되었다라고 생각했다.