백준 15651번 N과 M (3)

첫번째 풀이 :

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 _ in 0..<m {
            print(array.map{String($0)}.joined(separator: " "))
            return
        }
        
    }
    else {
        for i in 1...n {
            array.append(i)
            check(num: num+1)
            array.removeLast()
        }
    }
}

check(num: 0)

해당 문제에서는 true , false를 구분하지 않고 바로 진행하면 중복된 수도 포함된다.

 

하지만, 이렇게 제출할 경우 시간 초과가 뜨게 되는데, 이유를 찾아보니 print를 하는 부분에서 시간이 초과된다는 이유..

 

정답 :

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)
var result = ""

func check(num : Int) {
    if num == m {
        for _ in 0..<m {
            result += array.map{String($0)}.joined(separator: " ")
            result += "\n"
            return
        }
        
    }
    else {
        for i in 1...n {
            array.append(i)
            check(num: num+1)
            array.removeLast()
        }
    }
}

check(num: 0)
print(result)

따라서, 문자열에 저장한 후 마지막에 문자열을 출력하는 방법을 사용했다!