[C++][프로그래머스] N으로 표현 :: seoftware

 

문제 설명

 

아래와 같이 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의 최대 개수가 8개라는 점을 이용한다.


Dynamic programming (동적 프로그래밍 동적계획법)
을 사용하여 N이 1개일 때부터 8개일 때까지 가능한 숫자를 집합에 저장한다(중복방지를 위한 set 사용).


예를 들어, N이 3개 사용될 때는 N이 1개 사용되는 수와 N이 2개 사용되는 수 간의 사칙연산이 수행되거나 "NNN"인 경우가 있다.

 

 

소스코드

---

#include <string>
#include <vector>
#include <set>
#include <algorithm>
#include <cmath>
using namespace std;

int solution(int N, int number) {
    int answer = 0; 
    
    //"N"*(i+1)로 초기화
    vector<set<int>> arr(8);
    for(int i = 0; i < 8; i++){
        int n = 0;
        int digit = i+1;
        while(digit > 0){
            digit--;
            n += N*pow(10, digit);
        }
        arr[i].insert(n);
    }
    //[DP] 사칙연산으로 가능한 숫자 모두 확인
    for(int i = 1; i < 8; i++){
        for(int j = 0; j<i; j++){
            for(auto &x : arr[j]){
                for(auto &y : arr[i-j-1]){
                    arr[i].insert(x+y);
                    arr[i].insert(x*y);
                    arr[i].insert(x-y);
                    if(y!=0) arr[i].insert(x/y);
                }
            }
        }
        //number가 있는지 확인
        if(arr[i].find(number) != arr[i].end()){
            return i+1;
        }
    }
    //1부터 8까지 중에 return 없을시 -1 return
    return -1;
}