P1036 [NOIP2002 普及组] 选数

题目描述

已知 𝑛n 个整数 𝑥1,𝑥2,⋯ ,𝑥𝑛x1​,x2​,⋯,xn​，以及 11 个整数 𝑘k（𝑘<𝑛k<n）。从 𝑛n 个整数中任选 𝑘k 个整数相加，可分别得到一系列的和。例如当 𝑛=4n=4，𝑘=3k=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,k（1≤𝑛≤201≤n≤20，𝑘<𝑛k<n）。

第二行 𝑛n 个整数，分别为 𝑥1,𝑥2,⋯ ,𝑥𝑛x1​,x2​,⋯,xn​（1≤𝑥𝑖≤5×1061≤xi​≤5×106）。

输出格式

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

输入输出样例

输入 #1复制

4 3
3 7 12 19

输出 #1复制

1

代码：

#include <bits/stdc++.h>

using namespace std;
int n,k,ans=0;
int a[25];
int c[25];
int d[25];
int check(int num)
{
    bool flog=1;
    if(num==2)flog=1;
    for(int i=2;i<num;i++){
        if(num%i==0)flog=0;
    }
    if(flog==0) return 0;
    else return 1;
}
void dfs(int s)
{
    int num=0;
    bool b[25]={};
    if(s==k+1){
       for(int i=1;i<=s;i++) num+=c[i];
       if(check(num)==1) ans++;
       num=0;
       return;
    }
    for(int i=1;i<=n;i++){
        if(b[i]==1) continue;
        int flag=1;
        for(int j=1;j<s;j++)if(i<=d[j]) flag=0;
        if(flag==0) continue;
        d[s]=i;
        c[s]=a[i];
        b[i]=1;
        dfs(s+1);
        c[s]=0;
        b[i]=0;
        d[i]=0;
    }
}
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++) cin>>a[i];
    dfs(1);
    cout<<ans;
    return 0;
}