每日一题2月22日选数

P1036 [NOIP2002 普及组] 选数

题目描述：

题目给出n和k随后输入n个正整数，要求输出从这n个数中任选k个不同的数相加的和为素数的个数

数据范围：

1≤k<n≤20，正整数的范围不大于5*10e6

解题思路：

先从一般情况（暴力求解）的思路入手。

我们做过k固定的情况，比如x={1,2,3,4,5,6}，要求输出所有任意三个数组合的和。

const int n = 6;

int x[n]; for (int i = 1; i <= n; i++) x[i - 1] = i;
int k = 3;

for (int i1 = 0; i1 < n-(k-1); i1++)
	for (int i2 = i1+1; i2 < n-(k-2) ; i2++)
		for (int i3 = i+1; i3 < n-(k-k)); i3++)
			printf("%d\n",x[i1] + x[i2] + x[i3]);

可以看到, 我们需要k层for循环。第j次循环的范围是从  ，也就是说当k未知的时候，我们可以用递归的形式来解决，根据循环范围的规律来看

//第j轮循环,假设前面j-1个数字已经选好了  这一轮选第j个数
for(int i_j = i_(j-1); i_j < n-(k-j); i_j ++)
    f(i_j, j+1);

至少我们是需要两个参数：循环轮数和上一轮循环的循环变量；

但是如何存储已经选好的数?     那就再加一个sum参数

int q[20],count;

void f(int i,int j, int sum)//i=上一轮的循环变量也就是i_(j-1)假设sum=已选好的j个数之和
{
    if(j > k)
        {
            if(is_Prime(sum)) count++;
            return ;
        }

    for(int i = ; i < n-(k-j) ; i ++)
        f(i+1, j+1, sum + q[i]);
}

根据最终计算得出的sum来进行素数判断即可

代码：

#include<iostream>
#include<cstdio>

using namespace std;

const int N = 10e6 + 10;
int n, k, ans=0;
int q[20];

bool is_Prime(int x)
{
	if (x == 1) return false;
	if (x == 2)	return true;
	for (int i = 2; i * i <= x; i++)
		if (x % i == 0) return false;
	return true;
}
void f(int i, int j, int sum)
{
	if (j > k)
	{
		if (is_Prime(sum)) ans++;
		return;
	}

	for (int p = i; p < n - (k - j); p++)
		f(p+1, j + 1, sum + q[p]);
}
int main()
{
	scanf("%d%d", &n, &k);
	for (int i = 0; i < n; i++) scanf("%d", q + i);
	f(0, 1, 0);
	printf("%d", ans);
	return 0;
}