본문 바로가기

Algorithm/DFS

(C++) - 프로그래머스(고득점 kit - (DFS/BFS)) : 타겟 넘버(DFS) 답

반응형

programmers.co.kr/learn/courses/30/lessons/43165

 

코딩테스트 연습 - 타겟 넘버

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

programmers.co.kr

dfs로 푼 문제였습니다.

풀이방법

 1. 빼거나 더해서 depth == size까지  target과 같다면 1을 반환, 아니라면 0을 반환

 2. dfs함수 호출 후 cnt를 반환

 

 

Code

#include <bits/stdc++.h>

using namespace std;
int dfs(int depth, int num, vector <int> &numbers, int target){
    if(depth == numbers.size()){
        return num == target;
    }
    int cnt = 0;
    cnt += dfs(depth+1, num + numbers[depth], numbers, target);
    cnt += dfs(depth+1, num - numbers[depth], numbers, target);
    return cnt;
}

int solution(vector<int> numbers, int target) {
    return dfs(0,0, numbers, target);
}