쥐수의 공부노트

백준 15649번 N과 M (1) 본문

바킹독 알고리즘/백트래킹

백준 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

'바킹독 알고리즘 > 백트래킹' 카테고리의 다른 글

백준 1182번 부분수열의 합  (0) 2023.07.20
백준 9663번 N-Queen  (0) 2023.07.19