[Swift Algorithm] 두 개 뽑아서 더하기 Programmers(Lv.1)

문제 설명

정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요.

 

제한 조건

• numbers의 길이는 2 이상 100 이하입니다.
  • numbers의 모든 수는 0 이상 100 이하입니다.

 

입출력 예

numbers	result
[2,1,3,4,1]	[2,3,4,5,6,7]
[5,0,2,7]	[2,5,7,9,12]

 

내 문제 풀이

import Foundation

func solution(_ numbers:[Int]) -> [Int] {
    var result = [Int]()
    for i in 0..<numbers.count-1 {
        for j in i+1..<numbers.count {
            if !result.contains(numbers[i] + numbers[j]) {
                result.append(numbers[i] + numbers[j])
            }
        }
    }
    return result.sorted()
}

• 2중 for문을 사용하여 두 수를 더하는 반복문을 작성했고, result 배열에 같은 수가 있는지 확인하고 없다면 append 해주었음.
• 리턴문에서 정렬하여 반환.

 

인상적인 풀이

import Foundation

func solution(_ numbers:[Int]) -> [Int] {

    var arr:[Int] = []

    for i in 0..<numbers.count {
        for j in i+1..<numbers.count {
            arr.append(numbers[i]+numbers[j])
        }
    }

    return Set(arr).sorted()
}

• 중복 검사를 하지 않고 배열에 더한 수들을 저장, 리턴문에서 set를 이용하여 중복을 제거하고 sorted 해주었다.

 

💡 피드백

• Set 이용하니 굳이 중복 제거를 할 필요가 없다,,! 앞으로 Set를 잘 활용해봐야겠다.

 

---

 

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

func solution2(_ numbers:[Int]) -> [Int] {
    var res = Set<Int>()
    for i in 0..<numbers.count-1 {
        for j in i+1..<numbers.count { res.insert(numbers[i]+numbers[j]) }
    }
    return res.sorted()
}

• Set 사용하여 자동으로 중복 검사해주었고, 덕분에 첫 풀이보다 훨씬 간결해졌다.

 

---

 

문제

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