[프로그래머스] N으로 표현 (Java)

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

코딩테스트 연습 - 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번만 사용하여 표현할 수 있습니다.

 

풀이

DP로 분류된 문제지만 DFS로 풀었다.

사칙연산으로 만든 값이 number와 같을때 사용횟수의 최솟값이면 갱신하도록 했다.

사칙연산의 항으로 N, NN, NNN, ... 을 달아주도록 하였다. (nn = nn * 10 + 1으로 N을 NN으로 만들어줌)

for문의 범위를 [0,8)로 정한 이유 : N 으로 최대 8항 연산식을 만들 수 있음

코드

class Solution {
    public int answer = -1;
    // DFS (N, number, 사용 횟수, 만든값)
    public void dfs(int n, int num, int cnt, int val) {
        if(cnt > 8) return; // 사용 횟수가 8보다 크면 리턴 (answer = -1 유지)
        // 만든값이 number와 일치하고 사용 횟수가 최솟값이면 갱신
        if(num == val){
            if(answer > cnt || answer == -1) answer = cnt;
            return;
        }
        // N, NN, NNN, ... 에 대해 사칙연산 추가 (재귀)
        int nn = 0;
        for(int i = 0; i < 8; i++){
            nn = nn * 10 + n;
            dfs(n, num, cnt + 1 + i, val + nn);
            dfs(n, num, cnt + 1 + i, val - nn);
            dfs(n, num, cnt + 1 + i, val * nn);
            dfs(n, num, cnt + 1 + i, val / nn);
        }
    }
    
    public int solution(int N, int number) {
        dfs(N, number, 0, 0);
        return answer;
    }
}