[Swift] 두 정수 사이의 합

두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요. 예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.

제한 조건
* a와 b가 같은 경우는 둘 중 아무 수나 리턴하세요.
* a와 b는 -10,000,000 이상 10,000,000 이하인 정수입니다.
* a와 b의 대소관계는 정해져있지 않습니다.

Example

Input: a: 3, b: 5
Output: 12

Input: a: 3, b: 3
Output: 3

Input: a: 5, b: 3
Output: 12

풀러가기

코드

func solution(_ a:Int, _ b:Int) -> Int64 {
    if a == b {
        return Int64(a)
    }

    let minValue = min(a, b)
    let maxValue = max(a, b)
    return Int64((minValue ... maxValue).reduce(0, +))
}

풀이

요구사항대로 같은 경우 리턴으로 하도록 하고, Iterator 에서 max -> min 으로 가게되면 오류가 발생하기 때문에 min, max 함수를 사용했다.

다른 분의 멋진 코드

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

• TheVoid , 모모

잘 배웠습니다.

(3 + 5) * (max(3, 5) - min(3, 5) + 1) / 2
->
(3 + 5) * (5 - 3 + 1) / 2
->
8 * (2 + 1) / 2
->
8 * 3 / 2
->
24 / 2
->
12

수학적인 사고로 풀어서 놀랐다.
1 부터 n 까지의 합을 생각을 못한 게 아쉽다.