BOJ: #15655 - N과 M (6)

04:23

N과 M (1) ~ (5) 문제는 종만북에서 나온 코드를 활용해서 풀었다. (1)을 푼 코드로 (2)를 풀고, (3)을 풀고… 이런식으로 베이스 코드에서 전부 지우고 처음부터 풀지 않고 응용해나갔는데, (6)은 아예 베이스 코드를 지우고 풀었다.

문제

https://www.acmicpc.net/problem/15655

내가 작성한 코드

Swift

let input = readLine()!.split(separator: " ").map { Int(String($0))! }
let n = input[0]
let m = input[1]
let nums = readLine()!.split(separator: " ").map { Int(String($0))! }.sorted()
var picked = [String]()
/*
 n: 전체 원소의 수
 toPick: 더 고를 수 있는 원소의 수
 */
func pick(n: Int, toPick: Int, start: Int) {
    // 기저 사례: 더 이상 고를 수 있는 수가 없으면 출력하고 종료
    if toPick == 0 {
        print(picked.joined(separator: " "))
        return
    }
    for i in start..<n {
        picked.append(String(nums[i]))
        pick(n: n, toPick: toPick - 1, start: i+1)
        picked.removeLast()
    }
}

pick(n: n, toPick: m, start: 0)