[swift] 프로그래머스 - 연속된 수의 합

프로그래머스

---

문제풀이를 직관적으로 이해할 수 있도록 고차함수를 사용하지 않은 풀이와
고차함수를 사용한 풀이로 나누어 공부하고 있습니다.
- 문제에 따라 하나의 풀이만 있을 수 있습니다.

 

고차함수를 사용하지 않은 풀이

import Foundation

func solution(_ num:Int, _ total:Int) -> [Int] {
    var result = [total/num]
    var count = 1
    for i in 1..<num {
        if i % 2 != 0 {
            result.append(total/num + count)
        } else {
            result.append(total/num - count)
            count += 1
        }
    }
    return result.sorted()
}

• total / num 의 값을 연속된 수의 중심값으로 두고
• 반복문을 통해 중심값을 기준으로 두고 +1, -1, +2, -2... 차례대로 num의 값만큼 반복 횟수를 돌립니다.
• 해당 배열을 오름차순으로 정렬하면 원하는 result값이 나옵니다.

 

고차함수를 사용한 풀이

import Foundation

func solution(_ num:Int, _ total:Int) -> [Int] {
    var i = (total - (num * (1 + num) / 2)) / num
    return (1...num).map { $0 + i }
}

• 등차수열의 합공식을 사용해서 풀이했습니다.
• 등차수열의 합공식은 아래와 같습니다.

항의 개수 * (첫항 + 끝항) / 2

• total - (등차수열 합공식) / num 를 하면 우리가 원하는 배열의 첫번째 값보다 1이 작은 숫자가 나오게 됩니다.
• 그럼 항의 개수만큼 반복하며 각 값마다 첫번째 값을 더해서 배열로 리턴하면 원하는 result 배열이 나옵니다.