[문제]
n개의 음이 아닌 정수가 있습니다. 이 수를 적절히 더하거나 빼서 타겟 넘버를 만들려고 합니다. 예를 들어 [1, 1, 1, 1, 1]로 숫자 3을 만들려면 다음 다섯 방법을 쓸 수 있습니다.
사용할 수 있는 숫자가 담긴 배열 numbers, 타겟 넘버 target이 매개변수로 주어질 때 숫자를 적절히 더하고 빼서 타겟
넘버를 만드는 방법의 수를 return 하도록 solution 함수를 작성해주세요.
[내 풀이]
나는 위 문제를 아래 두가지 조건으로 접근해서 풀려고했다.
1. numbers 배열 모두를 사용해서 나올수 있는 모든 경우의 수(음수 포함)를 찾고
2. 각 경우의 수의 합이 target과 같은 경우를 카운팅
하지만 시간 초과로 통과하지 못했다. 효율성 측면이 문제인듯 한데 아마 1번 조건을 푸는 과정에서 시간이 많이 걸리는 듯 싶다..
import java.util.Iterator;
import java.util.LinkedHashSet;
import java.util.Set;
class Solution {
static Set<String> set = new LinkedHashSet<>();
public int solution(int[] numbers, int target) {
int answer = 0;
int[] arr = new int[numbers.length*2];
int[] output = new int[numbers.length*2];
boolean[] visited = new boolean[numbers.length*2];
int depth = 0;
int cnt = 0;
for (int i = 0; i < numbers.length; i++) {
arr[i] = numbers[i];
}
for (int i = numbers.length; i < arr.length; i++) {
arr[i] = -numbers[i-numbers.length];
}
// numbers 배열 모두 사용해서 나올수 있는 모든 경우의 수(음수 포함)를 찾고
perm(arr, output, visited, depth, arr.length, numbers.length);
// 각 경우의 수의 합이 target과 같은 경우를 카운팅
for (String item : set) {
int sum = 0;
boolean minus = false;
String[] s = item.split("");
for (String value : s) {
if (value.equals("-")) {
minus = true;
continue;
}
if (!minus) sum += Integer.parseInt(value);
else {
sum += -Integer.parseInt(value);
minus = false;
}
}
if (sum == target) answer++;
}
return answer;
}
private void perm(int[] arr, int[] output, boolean[] visited, int depth, int n, int r) {
if (depth==r){
StringBuilder sb = new StringBuilder();
for (int i = 0; i < r; i++) {
sb.append(output[i]);
}
set.add(sb.toString());
}
for (int i = 0; i < n; i++) {
if (!visited[i]){
visited[i] = true;
output[depth] = arr[i];
perm(arr, output, visited, depth+1, n, r);
visited[i] = false;
}
}
}
}
[다른 사람 풀이 참고]
dfs도 반드시 알고 있어야하는 알고리즘인 것 같다.
dfs를 구현하는 방법은 크게 재귀 방식과 스택을 이용한 방식이 있는데, 재귀를 이용한 방식을 참고하였다.
import java.util.Iterator;
import java.util.LinkedHashSet;
import java.util.Set;
class Solution {
int answer = 0;
public int solution(int[] numbers, int target) {
dfs(numbers, target, 0);
return answer;
}
private void dfs(int[] numbers, int target, int index) {
if (index == numbers.length) {
int sum = 0;
for (int num : numbers) sum += num;
if (sum == target)
answer++;
return;
}
dfs(numbers, target, index + 1);
numbers[index] *= -1;
dfs(numbers, target, index + 1);
}
}
<문제 출처>
'JAVA > Argorithm Test' 카테고리의 다른 글
# Level 2 - 소수 찾기 (0) | 2020.12.16 |
---|---|
# Level 2 - 전화번호 목록 (0) | 2020.12.15 |
# Level 2 - H-Index (0) | 2020.12.15 |
# Level 2 - 프린터 (0) | 2020.12.09 |
# Level 2 -124 나라의 숫자 (0) | 2020.12.08 |