洛谷-P1036 选数

洛谷-P1036 选数

原题链接：https://www.luogu.com.cn/problem/P1036

• 题目描述
• 输入格式
• 输出格式
• 输入输出样例
• C++代码

---

题目描述

已知n个整数(x_1,x_2,...,x_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。

输入格式

键盘输入，格式为：

n,k(1≤n≤20,k≤n)

(x_1,x_2,...,x_n)(1≤(x_i)≤5000000)

输出格式

屏幕输出，格式为： 1个整数（满足条件的种数）。

输入输出样例

输入 #1

4 3
3 7 12 19

输出 #1

1

C++代码

#include <iostream>
#include <cmath>
using namespace std;

int x[20],n,k;

bool isprime(int m){
    if(m==0||m==1)
        return false;
    int len=sqrt(m);
    for(int i=2;i<=len;++i)
        if(m%i==0)
            return false;
    return true;
}

int dfs(int h,int sum,int idx){
    if(h==0)
        return isprime(sum);
    int ans=0;
    for(int i=idx;i<n;++i)
        ans+=dfs(h-1,sum+x[i],i+1);
    return ans;
}

int main(){
    cin>>n>>k;
    for(int i=0;i<n;++i)
        cin>>x[i];
    cout<<dfs(k,0,0)<<endl;
}