바킹독 알고리즘/백트래킹
백준 15649번 N과 M (1)
쥐수
2023. 7. 12. 14:17
728x90
정답 :
let input = readLine()!.split(separator: " ").map{Int($0)!}
let n = input[0]
let m = input[1]
var array : [Int] = []
var issue = Array(repeating: false, count: n + 1)
func check(num : Int) {
if num == m {
for i in 0..<m {
print(array.map{String($0)}.joined(separator: " "))
return
}
}
else {
for i in 1...n {
if !issue[i] {
array.append(i)
issue[i] = true
check(num: num + 1)
issue[i] = false
array.removeLast()
}
}
}
}
check(num: 0)
아래의 코드 블럭은 주석을 달아 이해하기 쉽게 작성
let input = readLine()!.split(separator: " ").map{Int($0)!}
let n = input[0]
let m = input[1]
var array : [Int] = []
var issue = Array(repeating: false, count: n + 1)
func check(num : Int) { // 현재 num개까지의 수를 택했다.
if num == m { // m개를 모두 택했다면
for i in 0..<m {
print(array.map{String($0)}.joined(separator: " ")) // array에 저장된 숫자 출력
return
}
}
else {
for i in 1...n { // 1부터 n까지의 수에 대해
if !issue[i] { // 아직 i가 사용되지 않았으면
array.append(i) // 다음 숫자를 i로 정함
issue[i] = true // i를 사용되었다고 표시
check(num: num + 1) // 다음 수를 정하러 한 단계 더 들어감
issue[i] = false // num번째 수를 i로 정한 모든 경우에 대해 다 확인했으니 i를 이제 사용되지않았다고 명시함.
array.removeLast()
}
}
}
}
check(num: 0)
728x90