BOJ: #15656 - N과 M (7)

06:47

이 문제는 원래 toPick == 0일 때 picked를 바로 출력하게 했었는데, 이렇게 제출하면 시간초과가 난다. 그래서 혹시 하는 마음에, 문자열로 저장해서 한 번에 출력했더니 통과.

문제

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

내가 작성한 코드

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]()
var result = ""
/*
 n: 전체 원소의 수
 toPick: 고를 수 있는 원소의 수
 */
func pick(n: Int, toPick: Int) {
    // 기저 사례: 고를 수 있는 원소의 수가 없으면 출력하고 종료
    if toPick == 0 {
        result += picked.joined(separator: " ") + "\n"
        return
    }
 
    // 수를 고른다. 같은 수를 여러 번 골라도 된다.
    for i in 0..<n {
        picked.append(String(nums[i]))
        pick(n: n, toPick: toPick - 1)
        picked.removeLast()
    }
}
 
pick(n: n, toPick: m)
 
print(result, terminator: "")