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에 저장하여 한 번에 출력했다.