P1036 选数(递归)

题目描述

已知 nn 个整数 x_1,x_2,…,x_nx1​,x2​,…,xn​，以及11个整数kk(k<nk<n)。从nn个整数中任选kk个整数相加，可分别得到一系列的和。例如当n=4,k=3n=4,k=3,44个整数分别为3,7,12,193,7,12,19时，可得全部的组合与它们的和为：

3+7+12=223+7+12=22

3+7+19=293+7+19=29

7+12+19=387+12+19=38

3+12+19=343+12+19=34。

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3+7+19=293+7+19=29。

输入输出格式

输入格式：

 

键盘输入，格式为：

n,kn,k(1 \le n \le 20,k<n1≤n≤20,k<n)

x_1,x_2,…,x_n (1 \le x_i \le 5000000)x1​,x2​,…,xn​(1≤xi​≤5000000)

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

4 3
3 7 12 19

输出样例#1： 复制

1

#include<cstdio>
#include<cmath>//引入头文件 
using namespace std;
int n,k,a[21],s=0,ans=0;//定义全局变量，方便写函数 
bool f[21];//判断该数有没有被选过，用bool型变量 
int ss(int n)//定义判断素数的函数 
{
    if (n<=1) return 0;
	for (int i=2;i*i<=n;i++)
		if (n%i==0) return 0;
	return 1; 
}
int xs(int x,int y)//该函数是本程序中最关键的部分，认真看哦 
{//x为已经选了几个数，y为选第几个数 
    for(int i=y;i<=n;i+=1)//从y~n循环是为了避免重复的出现，例如1234中，选3个，已经选过123 
	{//与124时，准备为13选下一个数，此时若从1~n循环，则程序会先循环到2，又2此时为true，所以 //程序又会选2，所以123又再一次出现了，与先前所选出的123相重复

		if(f[i]==true)//如果该数没有被选过，则执行下列语句，反之，则i+=1或回溯
        {
            f[i]=false;//标志该数已经被选过 
            s+=a[i];//将相对应的值累加到s中 
            if(x==k)//如果已经加到了k个数，则运行下列语句 
            {
                if(ss(s))ans+=1;//判断和是否为素数，如果是素数，则ans+=1，标志情况多了一种 
            }
            else xs(x+1,i+1);//若还未加到k个数，则继续搜寻下一个数，所以x+1。i+1则是为了
            //搜寻当前数的下一个数，避免重复 
            s-=a[i];//回溯，累加器s减去a[i]的值 
            f[i]=true;//f[i]还原true，没有被选过
            //注意：两句回溯语句一定要放在else外面，x==k时无法继续往下搜寻，也需要回溯 
        }
    }
}
int main()
{
    scanf("%d%d",&n,&k);//读入共有几个数和每次选几个数 
    for(int i=1;i<=n;i+=1)
    {
        scanf("%d",&a[i]);//读入每个数的值 
        f[i]=true;//将判断该数选没选过的bool型数组初始化 
    }
    xs(1,1);//开始调用选数函数 
    printf("%d",ans);//输出结果 
    return 0;//结束程序 
}