[Level 1] 2016년 + Swift

2016년

코딩테스트 연습 - 2016년

나의 풀이

a = 5, b = 24로 5월 24일이라면 months배열의 0부터 4까지를 더하고 24를 더해 총 days를 구한다. 그 후 7로 나눈 나머지로 요일을 구한다.

" SAT"으로 앞에 띄어쓰기를 넣어서 테스트케이스 2와 10이 먹지 않아 당황했었다😫

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

    return weeks[count % 7]
}