【组合问题】-选数

[NOIP2002 普及组] 选数 - 洛谷

---

解题思路：

（1）读完题意，便可判断出这是一道组合的题目，要在n个数中选取m个数组合，在上节中已经具体实现如何操作

（2）本题略有不同的是，这n个数并不是1-n的排列，而是随机的整数，并且还添加了一个求和质数判断的过程，该怎么办呢？

（3）其实思路还是一样，可以这么想，将n个数字的下标看成1-n，选出来的组合会存放到ans数组中，那么ans【1】-ans【m】的值就是对应的原始数据的下标，即sum=sum+res【ans【i】】即可，如果还不理解，如下图：

 （4）最后设置一个判断质数的函数对sum进行判断即可，注意sum每次递归到出口，一定要清零，然后再累加！

---

#include<bits/stdc++.h>
using namespace std;
int n,m,sum,temp;
int ans[30],res[30];//ans数组存放组合的方案，res数组存放原始数据 
bool vis[30];//vis为监控数组 

bool judge(int x)//判断是否为质数 
{
	for(int i=2;i<=sqrt(x);i++)
	if(x%i==0)
	return false;
	
	return true;
}

void dfs(int x)
{
	if(x>m)//当取够的数字为m的时候，结束递归 
	{
		sum=0;//累加器为0 
		for(int i=1;i<=m;i++)//依次累加数字对应具体的值 
		{
			sum=sum+res[ans[i]];
		}
		
		if(judge(sum)==1)//如果和为质数 
		temp++;//计数器增加 
		
		return;
	}
	
	for(int i=1;i<=n;i++)//依次枚举可取的数字 
	{
		if(vis[i]==0&&i>ans[x-1])//当这个数字没有用过并且大于前一位数字的时候可用 
		{
			ans[x]=i;//将数字存放到ans数组中 
			vis[i]=1;//标记此数字已经被使用 
			dfs(x+1);//继续递归 
			vis[i]=0;//回溯撤销标记 
		}
	}
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	cin>>res[i];//输入原始数据 
	
	dfs(1);
	cout<<temp; //输出质数数量 
	return 0;
}