쥐수의 공부노트
백준 15649번 N과 M (1) 본문
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
'바킹독 알고리즘 > 백트래킹' 카테고리의 다른 글
백준 1182번 부분수열의 합 (0) | 2023.07.20 |
---|---|
백준 9663번 N-Queen (0) | 2023.07.19 |