k진수에서 소수 개수 구하기 C++

https://programmers.co.kr/learn/courses/30/lessons/92335

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

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

using namespace std;

bool ischeck(long long num) {
	if (num < 2) return false;

	for (int i = 2; i <= sqrt(num); i++) {
		if (num%i == 0) return false;
	}
	return true;
}

int solution(int n, int k) {
	int answer = 0;
	string value;

	while (n) {
		value += to_string(n%k);
		n = n / k;
	}
	
	string prime;
	for (int i = 0; i < value.size(); i++) {
		if ((value[i] == '0') && prime.size()>0) {
			reverse(prime.begin(), prime.end());
			if (ischeck(stoll(prime))) answer++;
			prime = "";
		}
		else if(value[i] != '0')
			prime += value[i];
	}

	if (prime.size() > 0) {
		reverse(prime.begin(), prime.end());
		if (ischeck(stoll(prime))) answer++;
	}

	return answer;
}