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

문제 링크

성능 요약

메모리: 4.2 MB, 시간: 0.01 ms

구분

코딩테스트 연습 > 동적계획법（Dynamic Programming）

채점결과

정확성: 100.0
합계: 100.0 / 100.0

문제 설명

아래와 같이 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 합니다.

 

---

#include <string>
#include <vector>
#include <set>

using namespace std;

int makeN(int N, int size)
{
  int answer = 0;
  for (int i = 0; i < size; i++)
    answer = answer * 10 + N;
  return answer;
}

int solution(int n, int num)
{
  vector<set<int>> dp;

  dp.push_back(set<int>());
  for (int i = 1; i <= 8; i++)
  {
    dp.push_back(set<int>());
    dp[i].insert(makeN(n, i));
    for (int j = 1; j <= i; j++)
    {
      for (auto x : dp[j])
      {
        for (auto y : dp[i - j])
        {
          dp[i].insert(x + y);
          dp[i].insert(x - y);
          dp[i].insert(x * y);
          if (y != 0)
            dp[i].insert(x / y);
        }
      }
    }
    if (dp[i].find(num) != dp[i].end())
      return (i);
  }
  return -1;
}