题目描述
已知 nn 个整数 x_1,x_2,…,x_nx1,x2,…,xn,以及11个整数kk(k<nk<n)。从nn个整数中任选kk个整数相加,可分别得到一系列的和。例如当n=4,k=3n=4,k=3,44个整数分别为3,7,12,193,7,12,19时,可得全部的组合与它们的和为:
3+7+12=223+7+12=22
3+7+19=293+7+19=29
7+12+19=387+12+19=38
3+12+19=343+12+19=34。
现在,要求你计算出和为素数共有多少种。
例如上例,只有一种的和为素数:3+7+19=293+7+19=29。
输入格式
键盘输入,格式为:
n,kn,k(1 \le n \le 20,k<n1≤n≤20,k<n)
x_1,x_2,…,x_n (1 \le x_i \le 5000000)x1,x2,…,xn(1≤xi≤5000000)
输出格式
屏幕输出,格式为: 11个整数(满足条件的种数)。
输入输出样例
输入 #1
4 3
3 7 12 19
输出 #1
1
#include <iostream>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#include <algorithm>
#include <set>
#include <map>
#include <climits>
#include <stdio.h>
#include <iomanip>
#include <math.h>
using namespace std;
int mycount = 0;
int size = 0;
bool isPrime(int number) {
for (int i = 2; i * i <= number; i++)
if (number % i == 0)
return false;
return true;
}
void backtrack(vector<vector<int>>& ret, vector<int> &tempList, vector<int>& nums, int start)
{
//ret.push_back(tempList);
if(tempList.size() == size)
{
int sum = 0;
for(int i = 0; i < size; i++)
{
sum += tempList[i];
}
if(isPrime(sum))
{
mycount++;
}
}
for(int i = start; i < nums.size(); i++)
{
tempList.push_back(nums[i]);
backtrack(ret, tempList, nums, i+1);
tempList.erase(tempList.begin() + tempList.size() - 1);
}
}
int main(void)
{
int n;
cin >> n >> size;
vector<int> nums(n);
for(int i = 0; i < n; i++)
{
cin >> nums[i];
}
vector<vector<int>> ret;
vector<int> tmp;
//sort(nums.begin(), nums.end());
backtrack(ret, tmp, nums, 0);
cout << mycount << endl;
return 0;
}