P1036 [NOIP2002 普及组] 选数

前言

  开始写dfs了...从以前就非常讨厌dfs...因为脑子里模拟不出这个过程，所以一直处理不好dfs...这一周估计都要写dfs了。

题目链接：P1036 [NOIP2002 普及组] 选数 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

一、题目

题目描述

已知 n 个整数 x1​,x2​,⋯,xn​，以及 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 个整数，分别为 x1​,x2​,⋯,xn​（1≤xi​≤5×106）。

输出格式

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

输入输出样例

输入 #1复制

4 3
3 7 12 19

输出 #1复制

1

说明/提示

【题目来源】

NOIP 2002 普及组第二题

二、分析

从题目给的数据范围也可以看出来了，这题可以用用暴力递归来解决。那么思路就很明确了，从n个数中取出m个数求和，并判断这样的和中有多少个素数。然后根据题意，很明显这里求和的数是没有顺序的（即1+2+4和1+4+2等等均为同一组）。对此，我们可以进行深搜得到素数的个数然后除以m的全排列得到正确的结果，也可以在深搜时控制搜索的顺序是非递减的，一样能得到正确的答案。以下是这两种方法的代码。

 

三、代码

第一种

#include<bits/stdc++.h>
using namespace std;
const int N=30;
int n,m;
bool book[N];//用来标记是否走过 
int a[N];
int cnt;//素数的个数 
int stu[N];
int A(int x){//用于计算全排列 
	int jc=1;
	for(int i=2;i<=x;i++){
		jc*=i;
	}
	return jc;
}
bool isprime(int x){
	if(x==1) return false; 
	for(int i=2;i<=sqrt(x);i++){
		if(x%i==0) return false;
	}
	return true;
}
void dfs(int step){
	if(step==m+1){//当step==m+1时，stu数组意味着已经放入了 m个数，可以进行求和判断是否为素数 
		int sum=0;
		for(int i=1;i<=m;i++){
			sum+=stu[i];
		}
		if(isprime(sum)){
			cnt++;
		}
		return ;
	}
	for(int i=1;i<=n;i++){//用深搜，列举出m个数的所有排列 
		if(book[i]==false){//序列第i个数还未走过 
			book[i]=true;
			stu[step]=a[i];
			dfs(step+1);
			book[i]=false;//搜到底，往回走，则第i个数要设置为没用过 （从这里就能看出，这样搜索出来的是有顺序的结果，如会包含1,2,3、1,3,2等等） 
		}
	}
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>a[i];
	dfs(1);
	cout<<cnt/A(m);
}

 第二种

#include<bits/stdc++.h>
using namespace std;
const int N=30;
int n,m;
int a[N];
int cnt;
bool isprime(int x){
	for(int i=2;i<=sqrt(x);i++){
		if(x%i==0) return false;
	}
	return true;
}
void dfs(int step,int sum,int start){//step与上一种方法是同含义，表明接下来要放入第step个数，sum为放入的数之和，start用来保证深搜数组的索引是非递减的 
	if(step==m+1){
		if(isprime(sum)){
			cnt++;
		}
		return ;
	}
	for(int i=start;i<=n;i++){//此处循环令i=start即为了保证深搜时，数组的索引是非递减的 （即只会有1,2,3一种情况，不会有2,3,1等等情况，故此方法不需要除以m的全排列) 
		dfs(step+1,sum+a[i],i+1);
		}
	}

int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>a[i];
	dfs(1,0,1);
	cout<<cnt;
}