동적계획법(Dynamic Programming) > N으로 표현 (Level 3)

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/42895

프로그래머스

 

문제 설명

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

 

입출력 예

 

입출력 예 설명

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

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

 

소스 코드

#include <string>
#include <vector>
#include <math.h>
#include <set>

using namespace std;

int MakeNumber( const int& N, const int& count )
{
	int ret = 0;
	for( int i = 0; i < count; ++i )
		ret += N * pow( 10, i );
	return ret;
}

int solution( int N, int number )
{
	const int maxSize = 8;
	std::vector< std::set< int >> dp;
	
	for( int i = 0; i < maxSize; ++i )
	{
		std::set< int > vec;
		vec.insert( MakeNumber( N, i + 1 ) );
		if( vec.find( number ) != vec.end() )
			return i + 1;

		for( int x = 0; x < i; ++x )
		{
			int y = i - x - 1;
			// 예외처리
			if( y < 0 )	continue;
		
			for( int num1 : dp[ x ] )
			{
				for( int num2 : dp[ y ] )
				{
					vec.insert( num1 + num2 );
					vec.insert( num1 - num2 );
					vec.insert( num1 * num2 );
					if( num2 != 0)
						vec.insert( num1 / num2 );
				}
			}
		}

		dp.emplace_back( vec );

		if( vec.find( number ) != vec.end() )
			return i + 1;
	}

	return -1;
}