C++에서 합 S가 있는 소수 P 이후의 소수

이 문제에서는 합 S, 소수 P, N의 세 가지 숫자가 주어집니다. 우리의 임무는 합이 S와 같은 P보다 큰 모든 N 소수를 찾는 것입니다.

문제를 이해하기 위해 예를 들어 보겠습니다.

Input: N = 2, P = 5, S = 18
Output: 7 11
Explanation: Prime numbers greater than 5 : 7 11 13
Sum = 7 + 11 = 18

이 문제를 해결하려면 P와 S 사이의 모든 소수를 찾아야 합니다. 그런 다음 합이 S가 되는 N개의 소수를 찾아야 합니다. 이를 위해 역추적을 사용합니다.

솔루션 구현을 보여주는 프로그램

예시

#include <iostream>
#include <vector>
#include <cmath>
using namespace std;
vector<int> set;
vector<int> primeNo;
bool isPrimeNumber(int x) {
   int sqroot = sqrt(x);
   bool flag = true;
   if (x == 1)
      return false;
   for (int i = 2; i <= sqroot; i++)
      if (x % i == 0)
         return false;
   return true;
}
void printPrimes() {
   int length = set.size();
   for (int i=0; i<length; i++)
   cout<<set[i]<<"\t";
   cout<<endl;
}
void GeneratePrimeSum(int total, int N, int S, int index) {
   if (total == S && set.size() == N) {
      printPrimes();
      return;
   }
   if (total > S || index == primeNo.size())
   return;
   set.push_back(primeNo[index]);
   GeneratePrimeSum(total+primeNo[index], N, S, index+1);
   set.pop_back();
   GeneratePrimeSum(total, N, S, index+1);
}
void PrimesWithSum(int N, int S, int P) {
   for (int i = P+1; i <=S ; i++) {
      if (isPrimeNumber(i))
      primeNo.push_back(i);
   }
   if (primeNo.size() < N)
   return;
   GeneratePrimeSum(0, N, S, 0);
}
int main() {
   int S = 23, N = 3, P = 3;
   cout<<N<<" Prime numbers greater than "<<P<<" with sum = "<<S<<" are :\n";
   PrimesWithSum(N, S, P);
   return 0;
}

출력

3 Prime numbers greater than 3 with sum = 23 are :
5   7   11