프로그래머스 Swift 타겟 넘버

문제

문제 설명

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

문제 풀이

재귀함수를 이용한 DFS(깊이우선탐색)으로 진행 해야한다. 처음에 다른 방식으로 접근하다가 구글링을 통해 알게 되었다.

• func dfs에서는 sum의 갯수의 합이 target의 값과 같다면 수를 증가 시켰다.
• 또한 리턴 이후 합에 dfs는 index의 수를 더하거나 빼주면서 재귀를 한다.

소스코드

import Foundation

var numbers = [1,1,1,1,1]
var target = 3

var number: [Int] = []
var targets: Int = 0
var count: Int = 0

func dfs(index: Int, sum: Int) {
    if index == number.count { //sum과 target의 계산값이 같다면 카운트 +1
        if sum == targets {
            count += 1
        }
        return //재귀가 가능하지 않을 경우 return
    }
    dfs(index: index + 1, sum: sum + number[index])
    dfs(index: index + 1, sum: sum - number[index])
}

func solution(_ numbers:[Int], _ target:Int) -> Int {
    number = numbers
    targets = target
    
    dfs(index: 0, sum: 0)

    return count
}