选数问题

选数问题

选数

【问题描述】

已知 n 个整数X₁,X₂,…,X_n,以及一个整数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₁,X₂,…,X_n(1≤X_I≤5000000)

【输出】

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

【样例输入】

4 3
3 7 12 19

【样例输出】

​ 1

思路分析

第一种思路

​ 要选择一些数是否符合条件，第一思路也是最容易想到的就是深搜，我的最初解法也是通过深搜，将所有数全部搜索一遍，看是否符合条件，符合条件数量加一。

​ 而其中有一些问题就是之前枚举过的情况如果不加以处理的话，后续还会计算这种情况，因为深搜就是将所有可能的情况全部列出，例如我已经计算过了3、7、12，之后程序可能又会计算一次7、3、12，12、3、7之类。所以我的解决思路是先将数据进行排序，此时数组内数据从小到大排列，dfs时额外加一个判断数组，判断一下当前所要枚举的数是否大于上一层所枚举的数，如果小于，说明这种情况之前已经列举过，继续循环枚举下个数，直至当前所要枚举的数大于上一层所枚举的数。这就是dfs的主要思路。

第二种思路

​ 这种思路是学长提供的，不得不说，我真的觉得这种思路非常巧妙，通过构造全排列来寻找我所选择哪些数。

#include<iostream>
#include<algorithm>
using namespace std;
int n,m;
int vis[205];
int main(){
	cin>>n>>m;
	for(int i=n-m+1;i<=n;i++){
		vis[i]=1;
	}
	do{
		for(int i=1;i<=n;i++){
			cout<<vis[i]<<' ';
		}
		cout<<endl;
	}while(next_permutation(vis+1,vis+1+n));
	return 0;
}

n=6,m=4所生成的全排列

通过一个vis数组，假如我n个数选m个数的话，从vis[n-m+1]开始初始为1，则vis中前n-m个数为0，通过不断的生成排列，能保证前后不会有两次相同的数，则对于原数组，我们取vis为1的对应的数组，这样就能保证运行时不会选择到重复的数。

这就是两种选数的思路，而关于判断素数呢，我选择的是六素数法（相关细节在这也不多说了，这个方法CSDN上有很多介绍），代码如下：

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

bool prime(int x){
	if(x<=1) return 0;
	if(x==2||x==3) return 1;
	for(int i=2;i<=sqrt(x);i++){
		if(x%i==0) return 0;
	}
	return 1;
}

bool six_prime(int x){
	if(x<=1) return 0;
	if(x==2||x==3) return 1;
	if(x%6!=1&&x%6!=5) return 0;
	//for(int i=5;i*i<=x;i+=6){
	//	if(x%i==0||x%(i+2)==0) return 0;
	//}
	int k=sqrt(x)+1;
	for(int i=6;i<=k;i+=6){
		if(x%(i-1)==0||x%(i+1)==0) return 0;
	}	
	return 1;
}

int main(){
	for(int i=1;i<=1000;i++){
		if(prime(i)) cout<<i<<' ';
		if(six_prime(i)) cout<<i<<'\n';
	}
	return 0;
} 

(左侧为原始的素数判断，右侧为六素数法，结果一样)

​ 而关于六素数法，其中有一些细节我也是一直一知半解，每次都想着用这种方法，每次都错，然后又用回原始方法，然后今天正式的对这个六素数做了下测试，总算发现了问题：

for(int i=2;i*i≤n;i++)//原始循环判断条件

int k=sqrt(x)+1;//六素数判断条件
for(int i=6;i<=k;i+=6)

​ 可以看出它是对x开方后加1进行条件判断，而之前一直使用原始方法的循环条件来进行判断，使得写一次错一次。

​ 这些大概的思路讲完后就是如下的代码了。

代码

#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int n,m,ans,tot;
int vis[2005],a[2005];

bool prime(int x){//判断质数
	if(x==1) return 0;
	if(x==2||x==3) return 1;
	if(x%6!=1&&x%6!=5) return 0;
	int k=sqrt(x)+1;
	for(int i=6;i<=k;i+=6){
		if(x%(i-1)==0||x%(i+1)==0) return 0;
	}
	return 1;
}

int main(){
	cin>>n>>m;
	
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=n-m+1;i<=n;i++)
		vis[i]=1;
	
	do{
		ans=0;
		for(int i=1;i<=n;i++){
			if(vis[i]) ans+=a[i];
		}
		if(prime(ans)) tot++;
	}while(next_permutation(vis+1,vis+n+1));
	cout<<tot;
	return 0;
}

（又进步了一点点，yeah)