[Programmers] k진수에서 소수 개수 구하기

문제 링크입니다: https://programmers.co.kr/learn/courses/30/lessons/92335

코딩테스트 연습 - k진수에서 소수 개수 구하기

문자열 처리를 통해 쉽게 풀 수 있는 문제였습니다.

 

	#include <iostream>
	#include <string>
	#include <vector>
	#include <algorithm>
	#include <cmath>
	using namespace std;
	string convert(int n, int k)
	{
	string result;
	while (n)
	{
	result += to_string(n % k);
	n /= k;
	}
	reverse(result.begin(), result.end());
	return result;
	}
	bool isPrime(string temp)
	{
	if (temp.empty() || temp == "1")
	{
	return false;
	}
	long long number = stol(temp);
	for (long long i = 2; i <= sqrt(number); i++)
	{
	if (number % i == 0)
	{
	return false;
	}
	}
	return true;
	}
	int solution(int n, int k) {
	string s = convert(n, k);
	string temp;
	int answer = 0;
	for (char c : s)
	{
	if (c == '0')
	{
	if (!temp.empty()
	&& isPrime(temp))
	{
	answer++;
	}
	temp = "";
	continue;
	}
	temp += c;
	}
	if (!temp.empty()
	&& isPrime(temp))
	{
	answer++;
	}
	return answer;
	}

view raw .cpp hosted with ❤ by GitHub

 

개발환경: Programmers IDE

 

지적, 조언, 질문 환영입니다! 댓글 남겨주세요~