[Swift] 백준 2309번 일곱 난쟁이

링크: https://www.acmicpc.net/problem/2309

2309번: 일곱 난쟁이

 

 

 

people로 9명의 난쟁이를 입력받습니다.

조합 함수로 총 7명의 난쟁이를 뽑고, 합계가 100이라면 결과를 업데이트합니다.

import Foundation

var people = [Int]()

for _ in 0..<9 {
    people.append(Int(readLine()!)!)
}

var result = [Int]()

func combination(_ index: Int, _ nowCombi: [Int]) {
    if nowCombi.count == 7 {
        if nowCombi.reduce(0, +) == 100 {
            result = nowCombi
        }
        return
    }
    for i in index..<people.count {
        combination(i + 1, nowCombi + [people[i]])
    }
}

combination(0, [])

result.sorted().forEach { print($0) }