Coding Test/programmers

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

owls 2023. 3. 16. 08:46
728x90

문제 설명

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

풀이

N의 개수에 따라 조합의 경우의 수가 증가합니다.

N 1개 : N  (1조합)

N 2개 : NN, N+N, N-N, N*N, N/N ( NN 1개와 N+-*/N  4개 총 5조합, 즉 이전 조합에 4개씩  증가됩니다. 
	 (1+1*4 = 5조합)
N 3개 : NNN, (NN+N), (NN-N), (NN*N), (NN/N), (N-N) ...
	(1+5*4 = 21조합)
N 4개 : NNNN, (NNN+N), (NNN-N), (NNN*N),(NNN.N) ... 
	(1+21*4 = 85조합)
.
.
.

조합되는 숫자 중에서 number가 있는지 확인하고 연산 횟수를 반환합니다. 

N이 최소로 사용된 값이어야 합니다.

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

using namespace std;

int solution(int N, int number) {
    int answer = -1;
    unordered_set<int> s[8];
    int sum = 0;
    for(int i = 0; i < 8; i++){
        sum = 10 * sum + N;		//N, NN, NNN, NNNN....조합을 만든다.
        s[i].insert(sum);
    }
    
    for(int i = 1; i < 8; ++i){
        for(int j = 0; j < i; ++j){
            for(int a : s[j]){
                for(int b : s[i-j-1]){	//이전 조합들을 가지고 각각 새로운 조합 생성
                    s[i].insert(a+b);
                    s[i].insert(a-b);
                    s[i].insert(a*b);
                    if( b != 0){
                        s[i].insert(a/b);
                    }
                }
            }
        }
    }
    
    for(int i = 0; i < 8; ++i){	//N 1개 사용부터 검사
        if(s[i].find(number) != s[i].end()){
            answer = i + 1;
            break;
        }
    }   
    return answer;
}
728x90