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

Seulguo·2022년 7월 12일
0

Algorithm

목록 보기
44/185
post-thumbnail
post-custom-banner

🐣 문제

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


🐥 코드

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

using namespace std;

int getNN(int N, int idx) {
	int result = N;
	for (int i = 1; i <= idx; i++) {
		result = result * 10 + N;
	}
	return result;
}

int solution(int N, int number) {
    if (N == number) return 1;
    
    vector<unordered_set<int>> dp(8);
    
    dp[0].insert(N);
    
    for(int k = 1; k < 8; k++){
        dp[k].insert(getNN(N, k));
        
        for(int i = 0; i < k; i++){
            for(int j = 0; j < k; j++){
                if(i + j + 1 != k) continue;
                
                for(int a : dp[i]){
                    for(int b : dp[j]){
                        dp[k].insert(a+b);
                        dp[k].insert(a*b);
                        if(a - b > 0) dp[k].insert(a-b);
                        if(a / b > 0) dp[k].insert(a/b);
                    }
                }
                
            }
        }
        if (dp[k].find(number) != dp[k].end()) return k+1;
    }
    return -1;
}
post-custom-banner

0개의 댓글