Algorithm/Baekjoon

[Swift] 백준 15649: N과 M(1)

내일은개발천재🎵 2022. 5. 8. 21:25

 부르트포스  

- 모든 경우의 수를 찾아 출력해야한다.

- 재귀함수를 이용하여 문제를 해결하였다.


  알고리즘  

1. N과 M을 입력받는다.
2. 1번부터 N번까지 다음과 같은 함수를 호출한다.
    a. 현재 선택된 번호를 방문처리한다. (중복 없도록 구현하기 위해)
    b. selected[i]에 현재 번호를 저장한다.
    c. 다음 번호로 이동하여 재귀함수를 호출한다.
    d. 호출 횟수가 M과 동일하다면, selected에 저장된 것을 result에 저장한다.
3. result를 출력한다.

let input = readLine()!.split(separator: " ").map{ Int(String($0))! }
let n = input[0], m = input[1]
var selected = [Int](repeating: 0, count:m)
var visited = [Bool](repeating: false, count: n+1)
var result = ""
rec_func(0)
print(result)


func rec_func(_ k: Int) {
    if k == m {
        result += selected.map{ String($0) }.joined(separator: " ") + "\n"
        return
    }
    for i in 1...n {
        if visited[i] { continue }
        selected[k] = i
        visited[i] = true
        rec_func(k+1)
        visited[i] = false
    }
}

  Key point!  

📍시간초과 오류를 해결하기 위한 방법으로 selected에 저장한 것을 바로 출력하지 않고, result에 저장하여 한 번에 출력했다.