[programmers/Swift] 두 정수 사이의 합

programmers.co.kr - 코딩테스트연습 - Lv.1 - 연습문제 - 두 정수 사이의 합

코딩테스트 연습 - 두 정수 사이의 합

Constraints :

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

 

Solution.swift :

//
//  Created by Yongwoo Marco on 2022/05/02.
//  Copyright © 2022 Yongwoo Marco Kim. All rights reserved.
//

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

print(solution(3, 5)) // 12
print(solution(3, 3)) // 3
print(solution(5, 3)) // 12

How I tried this :
첫 코드는 통과 되었지만 시간이 너무 많이 걸려서 다른 방법을 찾아보았다.

A...B 모든 수의 합도 좋지만 O(B-A) 만큼에 시간이 들기 때문에

 

A...B 형태는 곧 `(1...B-A).reduce(0) { $0 + $1 } * (B-A)`으로 볼 수도 있다

 

A < B 일때

(A + B) * (B - A + 1) / 2 공식이 두 사이의 모든 정수의 값이라는 걸 알아내서 적용했다.

 

What I got is :

왼쪽 (아래 코드) 오른쪽(위 코드)

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

문제에 관한 모든 저작권 : https://programmers.co.kr/