P1036 [NOIP2002 普及组] 选数

难度：普及-

题目描述

已知 n 个整数 x1​,x2​,⋯,xn​，以及 11 个整数 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。

输入格式

第一行两个空格隔开的整数 n,k（1≤n≤20，k<n）。

第二行 n 个整数，分别为 x1​,x2​,⋯,xn​（1≤xi​≤5×10^6）。

输出格式

输出一个整数，表示种类数。

输入输出样例

输入 #1

4 3
3 7 12 19

输出 #1

1

说明/提示

【题目来源】

NOIP 2002 普及组第二题

思路

先枚举全部的组合，再判断结果是否是素数

完整代码

#include <bits/stdc++.h>				//万能头文件 
using namespace std;					//标准命名空间 
int n,k,a[25];
int long long ans;
bool is(int a){							//定义bool函数标记素数 
    for(int i = 2; i * i <= a; i++){
    	if(a % i == 0)return false;
	}        
    return true;
}
void dfs(int m, int sum, int start){	m代表现在选择了多少个数，sum表示当前的和，start表示升序排列
    if(m == k){
        if(is(sum))ans++;				//如果和是素数，ans++ 
        return ;
    }
    for(int i = start; i < n; i++)
        dfs(m + 1, sum + a[i], i + 1);	//步数要加一，和也要加，升序起始值要变成i+1,以免算重
    return ; 
}
int main(){
    cin>>n>>k;							//输入 
    for(int i = 0; i < n; i++)cin>>a[i];//循环读入 
    dfs(0,0,0);							//调用函数 
    cout<<ans;							//输出结果 
    return 0;							//返回值为0 
}