C++で合計Sを持つ素数Pの後の素数

この問題では、合計S、素数P、およびNの3つの数が与えられます。私たちのタスクは、合計が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