[C++][프로그래머스] 타겟 넘버

프로그래머스

 

📄 문제

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 함수를 작성해주세요.

 

📝 풀이

#include <string>
#include <vector>
#include <iostream>

using namespace std;

bool ch[20];
int answer = 0;

void DFS(vector<int> numbers, int target, int p = 0)
{
    if(p == numbers.size())
    {
        for(int i = 0; i < numbers.size(); ++i)
            target += ch[i] ? +numbers[i] : - numbers[i];
        
        if(!target)
            ++answer;
    }
    else
    {
        for(int i = 0; i < 2; ++i)
        {
            ch[p] = i;
            DFS(numbers, target, p + 1);
        }
    }
}

int solution(vector<int> numbers, int target) {
    
    DFS(numbers, target);
    
    return answer;
}

• DFS를 이용하여 덧셈 및 뺄셈의 가능한 모든 조합을 ch에 저장하여 사용합니다.
• numbers의 개수만큼 조합을 구했다면, ch에 저장된 값을 통해 계산을 합니다.
• 계산식을 통해 target이 0이 되었다면, 조건에 맞음으로 answer의 개수를 1 올려줍니다.