영벨롭 개발 일지

[프로그래머스][C++]Level 2 타겟 넘버 풀이: DFS 본문

알고리즘 문제 풀이/프로그래머스

[프로그래머스][C++]Level 2 타겟 넘버 풀이: DFS

영벨롭 2022. 3. 8. 16:19

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

 

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

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

programmers.co.kr

 

[풀이 과정]

 

1. value는 해당 숫자, idx는 numbers 벡터에서 숫자의 위치, sum은 탐색하는 과정에서 해당 숫자까지의 합인 구조체 info 선언

2.  info 타입의 스택(벡터) 선언

3. {0, -1, 0}을 스택에 push 

4. 스택의 top 원소 pop

5. numbers 벡터에서 top원소 다음 위치에 해당하는 숫자를 양수, 음수로 스택에 push

6. top원소의 위치가 numbers의 마지막 위치이면서, 그 합이 target과 같다면 answer 증가

7. 스택이 empty일 때까지 4~6과정 반복

 

#include <string>
#include <vector>

using namespace std;

typedef struct info{
    int value;
    int idx;
    int sum;
}info;

int solution(vector<int> numbers, int target) {
    int answer = 0;
    
    vector<info> s;
    s.push_back({0, -1, 0});
    
    while(!s.empty()){
        info temp=s.back();
        s.pop_back();
        
        if(temp.idx==numbers.size()-1){
            if(temp.sum==target){
                answer++;
            }
            continue;
        }
        
        int val=numbers[temp.idx+1];
        int idx=temp.idx+1;
        int sum=temp.sum+val;
        
        s.push_back({val, idx, sum});
        
        val=val-val*2;
        sum=temp.sum+val;
        
        s.push_back({val, idx, sum});
        
    }
    
    return answer;
}
반응형