P1036 [NOIP2002 普及组] 选数

题目描述

已知 n个整数

​
以及1 个整数k(k<n)。从n 个整数中任选k 个整数相加，可分别得到一系列的和。
例如当n=4,k=3 4个整数分别为3,7,12,19 时，可得全部的组合与它们的和为：

3+7+12=22

3+7+19=29

7+12+19=38

3+12+19=34

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3+7+19=29 。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<stdio.h>
#include<math.h>
using namespace std;
long long int x[25], sum = 0;
int n, k;
int y[25] = {0}; //0为未读，1已读
int result = 0; //质数个数
void  fun(long long int p) { //判断是否未质数
	long long int i = 2;
	int b = 0;
	for ( ; i <= sqrt((double)p); i++) {
		if (p % i == 0 ) {
			b = 1;
			break;
		}
	}
	if (b == 0)
		result++;
}

void find(int i, int p ) { //访问了i个数，在访问第p个
	for (int j = p ; j <= n ; j++ ) { //j 从当前访问的p到最多的n
		y[j] = 1;       //已访问j，赋1 
		sum += x[j];
		//cout<<"x"<<j<<x[j]<<endl;
		if ( i == k )    //已访问的个数i 等于 要访问的个数k时 
			fun(sum);      //求和是否为质数 
		else find(i + 1, j + 1);   //未访问到规定次数时， 访问个数i+1，访问下一个 j+1 
		sum -= x[j];
		y[j] = 0;
	}
}
int main() {
	cin >> n >> k;
	for (int i = 1 ; i <= n ; i++)
		cin >> x[i];
	find(1, 1);
	cout << result << endl;
	return 0;
}