[Swift Algorithm] 두 정수 사이의 합 Programmers(Lv.1)

문제 설명

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

 

내 문제 풀이

func solution(_ a:Int, _ b:Int) -> Int64 {
    let largeN : Int
    let smallN : Int
    var result = 0
    
    if a > b {
        largeN = a
        smallN = b
    }
    else if b > a {
        largeN = b
        smallN = a
    }
    else {
        return Int64(a)
    }
    
    for i in smallN...largeN {
        result += i
    }
    
    return Int64(result)
}

• 입력 받은 두 수중 더 큰 수와 작은 수를 판별해주었고, 두 수가 같을 경우 바로 a를 출력, 같지 않을 때는 반복문으로 작은 수부터 큰 수까지 누적하여 출력해주었다.

 

인상적인 풀이

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

• Array, reduce 이용해서 푼 풀이를 발견했다,,! 완전 깔끔하다,,
• 근데 막상 돌려보니까 런타임 에러 잔뜩,,ㅜ 배열을 생성해서 한다는 것 자체가 시간을 너무 많이 잡아먹는 듯 하다.
• 그래도 Array 사용해서 만든 식이 배울 점이 많은 것 같아 가져와봤다.

 

💡 피드백

• 코드가 조금 간결하지 못한 것 같아서,, 더 좋은 방법을 찾아봐야겠다.!

 

---

 

🖍 [ 추가 ] 1주일 후 다시 풀어보기

func solution(_ a:Int, _ b:Int) -> Int64 {
    var result = 0
    
    if a > b {
        for i in b...a {
            result += i
        }
    }
    
    else {
        for i in a...b {
            result += i
        }
    }
    
    return Int64(result)
}

• a > b일 경우, 반대일 경우를 나눠서 계산해줬고, for문으로 누적한 값을 구해줬다.
• 첫 번째 풀이보다 훨씬 간결하고 가독성 측면에서도 좋은 듯 하다!

 

---

 

➕🖍 [ 또! 추가 ] 1주일 후 또 다시 풀어보기

func solution(_ a:Int, _ b:Int) -> Int64 {
    var sum = 0
    for i in (a > b ? b...a : a...b) { sum += i }
    return Int64(sum)
}

• 진짜 200배 깔끔해졌다..ㅜ 감격
• for문에서 범위로 조건식을 적을 수 있다는 것을 오늘 알게 되어서,, 한번 활용해봤다.!

 

---

 

 

문제

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