[c++] 프로그래머스 완전탐색 - 소수 찾기

알감자·2022년 5월 12일
0

프로그래머스

목록 보기
12/14

#소수찾기

문제

한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.

각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 solution 함수를 완성해주세요.

제한사항

  • numbers는 길이 1 이상 7 이하인 문자열입니다.

  • numbers는 0~9까지 숫자만으로 이루어져 있습니다.

  • "013"은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져있다는 의미입니다.

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

bool is_prime(int num)
{
    if(num==1 || num==0)
        return false;
    
    for(int i=2; i*i <= num; i++)
        if(num%i ==0)
            return false;
    
    return true;
}

int solution(string numbers) {
    vector<char> num;
	vector<int> temp;

	for (int i = 0; i<numbers.size(); i++)
	{
		num.push_back(numbers[i]);
	}

	sort(num.begin(), num.end());

	do
	{
		string str = "";
		for (int i = 0; i<num.size(); i++)
		{
			str += num[i];
			temp.push_back(stoi(str));
		}
	} while (next_permutation(num.begin(), num.end()));

	sort(temp.begin(), temp.end());

	temp.erase(unique(temp.begin(), temp.end()), temp.end()); 
    //중복제거

	int answer = 0;
	for (auto x : temp)
	{
		if (is_prime(x))
		{
			answer++;
		}
	}

    return answer;
}

0개의 댓글