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

---

 

나의 풀이

import Foundation

func solution(_ num:Int, _ total:Int) -> [Int] {
    for i in -500..<500 {
        if (i..<i+num).reduce(0, +) == total {
            return Array(i..<i+num)
        }
    }
    return []
}

- 반복문 범위도 잘못 됐고 return 도 이렇게 하면 안되겠쥬ㅋ

- 다른 테케 있었음 범위 때문에 실패했을거임^^

- 음수 범위는 어떻게 해결해야될까..?

 

 

다른 사람의 풀이 참고해서 다시 풀어봄

import Foundation

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

- 오 등차수열의 합 공식을 이용해서 풀 수 있군?

등차수열의 합 공식

n: 항 개수, a: 첫번째 항, l: 마지막 항

- 우리가 필요한건 a (첫번째 항)니까 그럼 a값을 알아낼 수 있는 식은

a = (total - (num * (num - 1) / 2)) / num

- 첫번째 항 a를 기준으로 0부터 num -1 순회해서 더해준 후 리턴하면 끝!