[프로그래머스 level 03] - 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 합니다.

입출력 예

Nnumberreturn

5	12	4
2	11	3

입출력 예 설명

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

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

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

입출력 예

Nnumberreturn

5	12	4
2	11	3

입출력 예 설명

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

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

 

---

내 풀이

 

일단 이 문제에서의 포인트는 문제 설명에도 힌트가 있듯이 DP로 풀어야한다.

그런데 나는 이해하는데 오래걸렸다 ㅠ 그래서 다른사람의 코드를 봤다 

 

가장 중요한 점은 

주어진 N을 1번부터 8번까지 사용된 수로 만들어서 배열이나 벡터에 넣는 것이었다 

(ex. 5, 55, 555, 5555, 55555, 555555, 5555555, 55555555)

 

그 뒤에 위의 수를 가지고 +, -, *, / 의 사칙 연산을 한 경우의 수의 결과값을 구해서 number와 일치하는지 보는 것이다.

그 중에 가장 적은 개수의 N을 쓴 횟수가 답이된다... 

 

또 하나 알아둘 것은 

+ 와 * 는 순서가 바뀌어도 값에 전혀 영향을 주지 않는다는 점이다. 다만, / 의 경우 나누는 수가 0일 경우에는 결과를 계산하면 안된다..

 

전체 소스 코드

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>

using namespace std;

int get_basic_number(int N, int cnt) {
	int res = 0;

	while (cnt > 0) {
		cnt -= 1;
		res += N * pow(10, cnt);
	}

	return res;
}

int solution(int N, int number) {
	int answer = -1;

	const int MIN = 8;

	vector<set<int>> s(MIN);

	int idx = 1;

	for (int i=0; i< s.size(); i++){
		s[i].insert(get_basic_number(N, idx));
		idx += 1;
	}

	for (int i = 1; i < MIN; i++) {
		for (int j = 0; j < i; j++) {
			for (const auto & op1 : s[j]) {
				for (const auto & op2 : s[i - j - 1]) {
					s[i].insert(op1 + op2);
					s[i].insert(op1 - op2);
					s[i].insert(op1 * op2);

					if (op2 != 0)
						s[i].insert(op1 / op2);
				}
			}
		}
		if (s[i].find(number) != s[i].end()) {
			answer = i + 1;
			break;
		}
	}
	return answer;
}

int main() {
	int N, number, result;

	result = solution(5, 12);
	cout << result;
}