프로그래머스 두 정수 사이의 합 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의 대소관계는 정해져있지 않습니다.

입출력 예

a	b	return
3	5	12
3	3	3
5	3	12

나의 풀이: a가 b보다 큰 경우와 a가 b보다 작은 경우 a와 b가 같은 경우 이렇게 3가지로 나눴다.

그리곤 var sum으로 합을 만들어주고 a,b값을 바꿀 수 없기에 각 각 newa,newb를 만들어주고 합에다가 더해준 뒤

a,b사이만큼 1씩 더해줬다.

import Foundation
func solution(_ a:Int, _ b:Int) -> Int64 {
    var sum = 0
    var newa = a
    var newb = b
    if a > b {
        for _ in 0...a-b{
        sum += newb
        newb += 1
        }
        return Int64(sum)
    } else if a < b{
        for _ in 0...b-a{
      sum += newa
       newa += 1
        }
        return Int64(sum)
 
    } else{
        return Int64(a)
    }
}
 

다른 사람의 풀이 중 가장 좋은 풀이

풀이 조건 ? a:b를 통해 값 추출

Array(a...b)를 이용해 한번에 a부터b까지의 값을 넣어버리고

reduce를 이용해 0부터 차례대로 더해버림

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