[Swift] 백준 - 부분 수열의 합

문제: https://www.acmicpc.net/problem/1182

1182번: 부분수열의 합

 

 

조합으로 풀었습니다

import Foundation

let arr = readLine()!.split(separator: " ").map { Int($0)! }
let N = arr[0]
let S = arr[1]

let array = readLine()!.split(separator: " ").map { Int($0)! }

var count = 0

func combi(_ nums: [Int], _ targetNum: Int) {
    func combination(_ index: Int, _ nowCombi: [Int]) {
        if nowCombi.count == targetNum {
            if nowCombi.reduce(0, +) == S {
                count += 1
            }
            return
        }
        for i in index..<nums.count {
            combination(i + 1, nowCombi + [nums[i]])
        }
    }
    
    combination(0, [])
}

for i in 1...N {
    combi(array, i)
}

print(count)