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

문제 설명

아래와 같이 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

 

 

코드

#include <string>
#include <vector>
#include <unordered_set>

using namespace std;

int solution(int N, int number) {
    unordered_set<int> us[9];
    int i, j, num;
    
    if (N == number)
    {
        return 1;
    }
    for (i = 1; i <= 8; i++)
    {
        num = 0;
        for (j = 0; j < i; j++)
        {
            num = num * 10 + N;
        }
        us[i].insert(num);
    }
    for (i = 2; i <= 8; i++)
    {
        for (j = 1; j < i; j++)
        {
            for (auto a : us[j])
            {
                for (auto b : us[i - j])
                {
                    us[i].insert(a + b);
                    us[i].insert(a - b);
                    us[i].insert(a * b);
                    if (b != 0)
                        us[i].insert(a / b);
                }
            }
        }
        if (us[i].find(number) != us[i].end())
        {
            return i;
        }
    }
    return -1;
}

 

링크

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

inbdni/Programmers