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

프로그래머스

---

문제풀이를 직관적으로 이해할 수 있도록 고차함수를 사용하지 않은 풀이와
고차함수를 사용한 풀이로 나누어 공부하고 있습니다.
- 문제에 따라 하나의 풀이만 있을 수 있습니다.

 

고차함수를 사용하지 않은 풀이

import Foundation

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

•  

 

고차함수를 사용한 풀이

import Foundation

func solution(_ numbers:[Int]) -> [Int] {
    return Set((0..<numbers.count).map { i in (i+1..<numbers.count).map { j in numbers[i] + numbers[j] } }.flatMap { $0 }).sorted()
}

• 과연 이게 의미가 있을까..?
• 그래도 변수를 하나 없앨 수 있어서 만족하고 있었는데 시간 복잡도를 보니

이건 for문 2개를 쓴 첫번째 풀이

이게 고차함수로 한줄로 푼 풀이

• 하마터면 통과 못할뻔.. 초과 뜨기 전에 겨우 통과 됐다.
• 좋은 알고리즘으로 코드를 짜는게 중요하다ㅎ..