개발을 시작하는 이야기

[프로그래머스] 두 정수 사이의 합 본문

개발 이야기/Algorithm Study

[프로그래머스] 두 정수 사이의 합

Teiresias 2022. 6. 2. 10:25

Swift Study 이주의 문제 3

 

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

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

programmers.co.kr

문제 설명

두 정수 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

Solution.swift

func solution(_ a:Int, _ b:Int) -> Int64 {
    var result = 0

    for i in a < b ? a...b : b...a {
        result += i
    }
    return Int64(result)
}

단순히 a < b인경우와 그렇지 않은 경우로 삼항연산자를 활용했다.


다른 사람의 풀이

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

깔끔 of 깔끔. 수열 공식을 활용해서 작성함

 

 

 

'개발 이야기 > Algorithm Study' 카테고리의 다른 글

[BaekJoon] 회의실 배정  (0) 2022.06.02
[LeetCode] Two Sum  (0) 2022.06.02
[BaekJoon] 주사위 세개  (0) 2022.06.02
[BaekJoon] 알람 시계  (0) 2022.06.02
[BaekJoon] 차량 번호판 1  (0) 2022.05.26