算法学习代码集

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

题目描述

已知 n 个整数 1,2,⋯ ,x1​,x2​,⋯,xn​，以及 11 个整数 k（<k<n）。从 n 个整数中任选 k 个整数相加，可分别得到一系列的和。例如当 n=4，k=3，44个整数分别为 3,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。

#include <iostream>

using namespace std;

const int N = 25;
int a[N];
int n, k, res;

bool is_prime(int x)
{
	for (int i = 2; i * i <= x; i++)
	{
		if (x % i == 0) return false;
	}
	return true;
}

void dfs(int cnt, int start, int sum)  // start表示起始位置, cnt为已选数的数量
{
	if (cnt == k)
	{
		if (is_prime(sum)) res++;

	return;
	}

	for (int i = start; i <= n; i ++)
	{
		dfs(cnt + 1, i + 1, sum + a[i]);
	}
	return;
}

int main()
{
	cin >> n >> k;

	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
	}

	dfs(0, 1, 0);

	cout << res;

	return 0;
}