[Swift] DFS - 타겟 넘버

Swift 로 프로그래머스 문제 풀기

1. 문제 설명

n개의 음이 아닌 정수가 있습니다. 이 수를 적절히 더하거나 빼서 타겟 넘버를 만들려고 합니다. 예를 들어 [1, 1, 1, 1, 1]로 숫자 3을 만들려면 다음 다섯 방법을 쓸 수 있습니다.

-1+1+1+1+1 = 3
+1-1+1+1+1 = 3
+1+1-1+1+1 = 3
+1+1+1-1+1 = 3
+1+1+1+1-1 = 3
사용할 수 있는 숫자가 담긴 배열 numbers, 타겟 넘버 target이 매개변수로 주어질 때 숫자를 적절히 더하고 빼서 타겟 넘버를 만드는 방법의 수를 return 하도록 solution 함수를 작성해주세요.

제한사항
주어지는 숫자의 개수는 2개 이상 20개 이하입니다.
각 숫자는 1 이상 50 이하인 자연수입니다.
타겟 넘버는 1 이상 1000 이하인 자연수입니다.
입출력 예
numbers target return
[1, 1, 1, 1, 1] 3 5
입출력 예 설명
문제에 나온 예와 같습니다.

2. 나의 풀이

func solution(_ numbers:[Int], _ target:Int) -> Int {
    var ch: [Int] = []
    var ret: Int = 0
    var ans: Int = 0
    var j: Int = 0
    func D(_ n: Int) {
        if ch.count == numbers.count {
            ret = 0
            for i in 0..<numbers.count {
                if ch[i] == 1 {
                    ret += numbers[i]
                } else {
                    ret -= numbers[i]
                }
            }
            if ret == target {
                ans += 1
            }
        } else {
            ch.append(1)
            D(n + numbers[j])
            ch.removeLast()
            ch.append(0)
            D(n - numbers[j])
            ch.removeLast()
        }
    }
    D(0)
    return ans
}

3. 풀이 설명

재귀함수를 처음 사용해보았다.
이러한 문제풀이를 깊이 우선 탐색 이라고 부른다.
영어로는 Depth - First - Search DFS

일단 check 배열을 만든다 ( 풀이에는 ch )
check 에는 numbers의 숫자가 + 숫자 이면 1, 아니면 0 이 들어간다.
(처음에는 0 도 아니고 빈 배열이 들어가있다. 이유는 아래에)
그러고 맨 마지막에 ch[i] == 1 이면 numbers[i] 를 더하고 0이면 빼준다
그리고 이 마지막 sum 하는 과정은
배열의 크기가 5 일때 실행한다.! 그래서 처음엔 빈 배열을 넣어줬다.

4. 결과

100점!