PTA 选数

最新推荐文章于 2024-10-12 22:21:30 发布

真正的大师怀着一颗学徒的心

最新推荐文章于 2024-10-12 22:21:30 发布

阅读量261

点赞数 1

文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/qq_74818985/article/details/130241442

版权

给定n个整数和一个整数k，需要通过DFS遍历所有可能的k个数的组合，计算其和，如果和为素数，则计数加一。最后输出和为素数的组合总数。题目中给出了一个示例和代码实现，代码使用C++编写，主要利用了DFS进行深度优先搜索寻找组合。

摘要由CSDN通过智能技术生成

7-5 选数

已知n个整数x1,x2,x3...xi，以及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）
第二行n个整数，两数之间空格隔开（1≤xi≤1000000）

输出格式:

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

输入样例:

在这里给出一组输入。例如：

4 3
3 7 12 19

输出样例:

在这里给出相应的输出。例如：

代码长度限制

16 KB

时间限制

1000 ms

内存限制

128 MB

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=50;
int a[maxn];
int n,k,ans;
bool prime(int x){//判断是不是素数
	for(int i=2;i<=sqrt(x);i++){
		if(x%i==0)return false;
	}
	return true;
}
void dfs(int t,int sum,int d){
    if(t==k){
        if(prime(sum))ans++;//达到k个 检验是否是素数 计数
    }
    else{
        for(int i=d;i<n;i++){//没到k个  继续从后边找下一个数
            dfs(t+1,sum+a[i],i+1);
        }
    }
}
int main(){
    cin>>n>>k;
    for(int i=0;i<n;i++){
    cin>>a[i];
}
    dfs(0,0,0);
    cout<<ans<<endl;
    return 0;
}