[Swift｜코드카타] (프로그래머스) 입문 #22. 두 정수 사이의 합 - Array(), 등차수열의 합

🔴 문제

두 정수 사이의 합 문제 (클릭)

 

🔵 풀이 1 - for...in

func solution(_ a: Int, _ b: Int) -> Int {
    var sum = 0
    let start = min(a, b)
    let fin = max(a, b)
    
    for i in start...fin {
        sum += i
    }
    return sum
}

* for i in start...fin 에서 star는 무조건 fin보다 작거나 같아야 한다.

 

🔵 풀이 2 - Array함수 사용

func solution(_ a:Int, _ b:Int) -> Int64 {
    return Int64(Array(a > b ? b...a : a...b).reduce(0, +))
  }

 

Array 함수에 대한 문서는 못찾았지만, Array(a...b)는 a부터 b까지의 정수값들이 포함된 리스트를 반환한다.

Array 함수

 

🔵 풀이 3 - 등차수열의 합 공식 사용

1부터 n까지 모든 정수의 합은 n(n+1)/2로 계산할 수 있다.

func solution(_ a:Int, _ b:Int) -> Int64 {
    return Int64(a + b) * Int64(max(a, b) - min(a, b) + 1) / Int64(2)
}