[프로그래머스] 두 개 뽑아서 더하기

<문제>

이 문제는 정수형 배열이 주어지고, 그 안에서 임의의 두 숫자를 골라 두 수를 합한 값의 모든 경우를 구해야 하는 문제다.

 

---

<전략짜기>

제한사항에 주어지는 배열의 크기를 보면 Worst Case의 탐색 횟수는 10000회 (100x100) 인데, 사실상 이정도는 제한시간에 영향을 미치지 않는다고 판단했다. (1억회 연산/탐색 기준 1초로 판단)

그냥 Brute force 방식으로 배열을 탐색하면서 두 수의 합을 모은 뒤에 마지막에 중복제거, 정렬만 해주면 되겠구나.

정렬은 Array에서 제공하는 .sorted() 메소드를 사용하고, 중복제거는 Set을 이용하면 되겠는걸..!?

 

<코드>

import Foundation

var result: [Int] = []

func solution(_ numbers:[Int]) -> [Int] {
    for i in 0 ..< numbers.count {
        for j in 1 ..< numbers.count {
            if i == j {
                continue
            } 
            result.append(numbers[i] + numbers[j])
        }
    }
    
    let resultSet = Set(result)
    let resultArray = Array(resultSet)
    result = resultArray.sorted()

    return result
}

 

<Lessons Learned>

정답으로 통과하긴 했지만 다른사람의 풀이를 보고 아차 싶었다 ㅎㅎ

나는 "배열에서 중복제거를 위해서 중복을 허용하지 않는 자료구조 중 하나인 Set을 이용하면 쉽겠구나." 하고 쉽게 생각한 것 같다.

이렇게 하는 방법 대신, result 배열에 두 수의 합의 경우를 넣어주는 과정에서 .contains() 메소드로 바로바로 중복체크를 해줬으면 더 좋았을 듯 하다..

if !result.contains(num) {
	result.append(num)
}

불필요한 Array -> Set -> Array 로의 변경에 대한 메모리/시간을 이득을 볼 수 있었을텐데. 😂

 

 

알고리즘에서 사용한 Set 내용이 궁금하다면 ? 👉 2020/12/15 - [iOS/기초를 탄탄히!] - [Swift] Set

 

 

# 출처 : 프로그래머스 두 개 뽑아서 더하기