프로그래머스 - 소수 찾기

문제 : https://programmers.co.kr/learn/courses/30/lessons/42839

코딩테스트 연습 - 소수 찾기 | 프로그래머스

 

 

풀이 :

 

picked 배열은 numbers의 1부터 numbers의 길이까지 숫자들을 쪼개서 넣게 된다.

 

예를 들면 17을 넣게 되면 1 / 7 / 1,7 / 7,1을 담게 된다.

 

이제 각 후보들을 구했다면 이 후보들 중 소수를 찾자. 중복된 수가 나 올 수 있기에 set을 통해 중복

 

을 없애주자.

 

코드(C++)

	#include <string>
	#include <vector>
	#include <iostream>
	#include <set>
	using namespace std;
	bool visited[8];
	int ret = 0;
	set<int> ss;
	void solve(int n, vector<char>& picked, vector<char>& v)
	{
	if (picked.size() == n)
	{
	string temp = "";
	for (int i = 0; i < picked.size(); ++i)
	{
	temp += picked[i];
	}
	int ans = stoi(temp);
	bool check = true;
	if (ans == 1 || ans == 0) return;
	for (int i = 2; i < ans; ++i)
	{
	if (ans%i == 0) {
	check = false;
	break;
	}
	}
	if (check) {
	if (ss.find(ans) != ss.end())
	return;
	ret++;
	ss.insert(ans);
	}
	return;
	}
	for (int i = 0; i < v.size(); ++i)
	{
	if (visited[i]) continue;
	picked.push_back(v[i]);
	visited[i] = true;
	solve(n, picked, v);
	picked.pop_back();
	visited[i] = false;
	}
	}
	int solution(string numbers) {
	int answer = 0;
	vector<char> v;
	for (int i = 0; i < numbers.size(); ++i)
	{
	v.push_back(numbers[i]);
	}
	vector<char> picked;
	for (int i = 1; i <= numbers.size(); ++i)
	{
	solve(i, picked, v);
	}
	return ret;
	}

view raw .cpp hosted with ❤ by GitHub