프로그래머스 - 타겟 넘버(Level 2)/Wanna Be 컴잘알

문제 출처 - https://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+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 이하인 자연수입니다.

 

- 접근 -

 

DFS를 사용해서 음수로 바꿀 수를 0개~numbers.size()-1 만큼 고르고 모든 경우의 수를 찾는다.

(numbers.size()만큼 고를 필요는 없는것이 target넘버가 양수라고 했으니 모두 음수인 경우는 제외하고 dfs를 돌린다.)

각 경우의 수마다 계산을 했을 때, 계산한 값이 target넘버가 되는지 체크를 해주면된다.

 

- 코드 -

 

 c++

#include <string>
#include <vector>
#include <string.h>

using namespace std;
bool check[20];
int answer = 0;

void solve(vector<int> numbers, int target, vector<int> index, int end, int cnt, int standard, vector<int> &v) {
	if (end == cnt) { //음수로 바꿔야할 목표 개수에 도달했을 때,
		int sum = 0;
		bool cal[20]; //음수로 바꾼 자리가 어딘지 알기위해
		memset(cal, false, sizeof(cal));


		for (int k = 0; k < v.size(); k++) {
			cal[v[k]] = true;
			sum -= numbers[v[k]]; //음수로 바꾸어더한다.
		}
		
		for (int k = 0; k < numbers.size(); k++) {
			if (cal[k] == false) {
				sum += numbers[k]; //나머진 그냥 더한다.
			}
		}

		if (sum == target) { //타겟넘버와 같을 경우 답에 +1 추가
			answer += 1;
		}
		return;
	}
	for (int k = standard; k < index.size(); k++) {
		if (!check[k]) {
			check[k] = true;
			v.push_back(index[k]);
			solve(numbers, target, index, end, cnt + 1, k, v);
			v.pop_back();
			check[k] = false;
		}
	}
}

int solution(vector<int> numbers, int target) {
	answer = 0;
	vector<int> index; // 인덱스를 저장하기 위한 벡터
	memset(check, false, sizeof(check));
	
	int sum = 0;

	for (int i = 0; i < numbers.size(); i++) {
		sum += numbers[i];
	}

	if (sum == target) //다 양수로 처리해서 타겟넘버가 될 경우
		return 1;

	for (int i = 0; i < numbers.size(); i++) {
		index.push_back(i);
	}

	for (int i = 1; i < numbers.size(); i++) {
		vector<int> v;
		solve(numbers, target, index, i,0,0, v);
	}
	
	
	return answer;
}

 

python 3 (추가 업로드)

 

# dfs sovle로 해결 
check=[]
vector=[]
answer = 0

def cal(target,numbers):
    global vector,answer
    temp = [i for i in numbers]
    
    for i in vector:
        temp[i] *=-1
    
    sum=0
    for i in temp:
        sum+=i
    
    if target == sum:
        answer+=1

def solve(cnt, standard, end, numbers,target):
    global check, vector

    if cnt == end:
        cal(target,numbers)
        return
    
    for i in range(standard, len(numbers)):
        if check[i] == False:
            check[i] = True
            vector.append(i)
            solve(cnt+1,i,end,numbers,target)
            vector.pop()
            check[i]=False

def solution(numbers, target):
    global check, answer

    for k in range (1,len(numbers),1):
        vector = []
        check = [False for i in range (0,20,1)]
        solve(0,0,k,numbers,target)

    return answer