두 정수 사이의 합

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12912

프로그래머스

 

 

내 답안 / 다른 사람들의 답안

func solution(_ a:Int, _ b:Int) -> Int64 {
    return Int64(([a, b].min()!...[a, b].max()!).reduce(0, +))
}

- 두 수의 대소관계가 정해져있지 않아 둘 중 작은 수부터 큰 수 사이의 배열을 만들고 그것을 다 더해서 반환하는 방법을 사용했다.

- 일부 테스트 케이스에서 시간이 많이 소요되어 최적화가 필요함을 느꼈다.

 

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

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

- 풀이 1은 내 답안과 똑같이 시간이 많이 걸리는 풀이이고, 풀이 2가 주목할만했다.

- 등차수열의 합 공식인 가우스의 합을 사용한 풀이로, 등차수열의 합은 항의 개수 * (첫째 항 + 마지막 항) / 2 로 구할 수 있다는 것이다.

- Int64(a + b) = (첫째 항 + 마지막 항), Int64(max(a, b) - min(a, b) + 1) = 항의 개수

 

점수: +1