[프로그래머스/DP] N으로 표현

문제 설명

아래와 같이 5와 사칙연산만으로 12를 표현할 수 있습니다.

12 = 5 + 5 + (5 / 5) + (5 / 5)
12 = 55 / 5 + 5 / 5
12 = (55 + 5) / 5

5를 사용한 횟수는 각각 6,5,4 입니다. 그리고 이중 가장 작은 경우는 4입니다.
이처럼 숫자 N과 number가 주어질 때, N과 사칙연산만 사용해서 표현 할 수 있는 방법 중 N 사용횟수의 최솟값을 return 하도록 solution 함수를 작성하세요.

 

제한사항

• N은 1 이상 9 이하입니다.
• number는 1 이상 32,000 이하입니다.
• 수식에는 괄호와 사칙연산만 가능하며 나누기 연산에서 나머지는 무시합니다.
• 최솟값이 8보다 크면 -1을 return 합니다.

 

입출력 예

N number return

5	12	4
2	11	3

 

입출력 예 설명

예제 #1
문제에 나온 예와 같습니다.

예제 #2
11 = 22 / 2와 같이 2를 3번만 사용하여 표현할 수 있습니다.

 

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

코딩테스트 연습 - N으로 표현

 

---

내 소스

import java.util.*;

class Solution {
    TreeSet<Integer>[] dp;
    int number;
    public int solution(int N, int number) {
        this.number = number;
        dp = new TreeSet[10];
        /*
        dp 값 초기 세팅
        dp[0] : {0}
        dp[1] : {5}
        dp[2] : {55}
        ..
        dp[8] : {55555555}
        */
        dp[0] = new TreeSet<>();
        dp[0].add(0);
        for(int i=1; i<=8; i++){
            dp[i] = new TreeSet<>();
            dp[i].add(dp[i-1].first() * 10 + N);
            if(dp[i].first() == number){
                return i;
            }
        }
        /*
        N이 5인 경우
        dp[0] = {0}
        dp[1] = {5}
        dp[2] = {55, 5+5, 5-5, 5*5, 5/5} = {55, 10, 0, 25, 1}

        뺄셈을 예로 들면
        dp[3] : dp[1]-[2], dp[2]-dp[1]
        dp[4] : dp[1]-dp[3], dp[2]-dp[2], dp[3]-dp[1]   
        dp[5] : dp[1]-dp[4], dp[2]-dp[3], dp[3]-dp[2], dp[4]-dp[1]
        */
        for(int i=2; i<=8; i++){
            for(int j=1; j<i; j++){
                int k = i-j;
                dp(i, j, k);
            }
            
            if(dp[i].contains(number)){
                return i;
            }
        }
        return -1;
    }
    
    public void dp(int cur, int i, int j){
        Iterator<Integer> iter = dp[i].iterator();
        int x;
        int y;
        while(iter.hasNext()){
            x = iter.next();
            Iterator<Integer> iter2 = dp[j].iterator();
            while(iter2.hasNext()){
                y = iter2.next();
                dp[cur].add(x+y);
                dp[cur].add(x-y);
                dp[cur].add(x*y);
                if(y != 0){
                    dp[cur].add(x/y);
                }
            }
        }
    }
}