JAVA/Argorithm Test

# Level 2 - 타겟 넘버

skysoo1111 2020. 12. 18. 11:09

[문제]

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);
    }
}

 

<문제 출처>

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

'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