BOJ: #15651 - N과 M (3)

종만북의 예시를 참고하여 풀고 있는데, 동일한 코드임에도 불구하고 스위프트에서는 시간 초과, 파이썬에서는 통과되는 기묘한 현상이 있었다. 원래 코드는 매개 변수로 picked 배열을 inout으로 받고 있었는데, 이것이 문제였다. 도움을 주신 택현님께 이 자리를 빌어 감사의 말씀을 드린다. 🙏 원인 찾는다고 1시간을 헤맸다. 흑흑

문제

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

내가 작성한 코드

Swift

let input = readLine()!.split(separator: " ").map { Int(String($0))! }
let n = input[0]
let m = input[1]
 
var picked = [Int]()
/*
 n: 전체 원소의 수
 picked: 지금까지 고른 원소들의 번호
 toPick: 더 고를 원소의 수
 일 때, 앞으로 toPick개의 원소를 고르는 모든 방법을 출력한다
 */
func pick(n: Int, picked:inout [Int], toPick: Int) {
    // 기저 사례: 더 고를 원소가 없을 때 고른 원소들을 출력한다.
    if toPick == 0 {
        for p in picked {
            print(p, terminator: " ")
        }
        print()
        return
    }
    // 원소 하나를 고른다.
    for next in 1...n {
        picked.append(next)
        pick(n: n, picked: &picked, toPick: toPick - 1)
        picked.removeLast()
    }
}
 
pick(n: n, picked: &picked, toPick: m)

Python

import sys
n, m = map(int, sys.stdin.readline().split())
picked = []
 
"""
 n: 전체 원소의 수
 picked: 지금까지 고른 원소들의 번호
 toPick: 더 고를 원소의 수
 일 때, 앞으로 toPick개의 원소를 고르는 모든 방법을 출력한다
"""
def pick(n: int, picked: [int], toPick: int):
    # 기저 사례: 더 고를 원소가 없을 때 고른 원소들을 출력한다.
    if toPick == 0:
        print(*picked)
        return
    # 원소 하나를 고른다.
    for next in range(1, n+1):
        picked.append(next)
        pick(n, picked, toPick - 1)
        picked.pop()
 
pick(n, picked, m)