[백준] N과 M (4) (15652)

문제 설명

백준 15652번 문제 링크

입력 및 출력

» 입력

첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

» 출력

• 한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해야 한다.

예제 입출력(테스트케이스)

입력	출력
3 1	1 2 3
4 2	1 1 1 2 1 3 1 4 2 2 2 3 2 4 3 3 3 4 4 4
3 3	1 1 1 1 1 2 1 1 3 1 2 2 1 2 3 1 3 3 2 2 2 2 2 3 2 3 3 3 3 3

문제 풀이(SWIFT) 1

//
//  main.swift
//  BOJ15652_SWIFT
//
//  Created by choiyoujun on 2022/02/18.
//

let NM = readLine()!.split(separator: " ").map { Int($0)! }
let N = NM[0]
let M = NM[1]
var stack = [Int]()
var ans = ""

func result(depth: Int) -> Void {
    if depth == M {
        ans += stack.map { String($0) }.joined(separator: " ")
        ans += "\n"
        return
    } else {
        if stack.isEmpty {
            for i in 1...N {
                stack.append(i)
                result(depth: depth + 1)
                stack.removeLast()
            }
        } else {
            for i in 1...N {
                if stack[depth - 1] <= i {
                    stack.append(i)
                    result(depth: depth + 1)
                    stack.removeLast()
                }
            }
        }
    }
}

result(depth: 0)
print(ans)