[swift] 프로그래머스 - 삼총사

프로그래머스

---

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

 

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

import Foundation

func solution(_ number:[Int]) -> Int {
    var count = 0
    for num1 in 0..<number.count {
        for num2 in num1+1..<number.count {
            for num3 in num2+1..<number.count {
                if number[num1] + number[num2] + number[num3] == 0 {
                    count += 1
                }
            }
        }
    }
    return count
}

•  

 

고차함수를 사용한 풀이

import Foundation

func solution(_ number:[Int]) -> Int {
    var count = 0
    (0..<number.count).forEach { num1 in (num1+1..<number.count).forEach { num2 in (num2+1..<number.count).forEach { if number[num1] + number[num2] + number[$0] == 0 { count += 1 } } }}
    return count
}

•