洛谷P1036 选数

#include <iostream>
#include <cmath>
using namespace std;
long long ans;
int n, k;
int a[25];
bool isprime(int a) {
	for (int i = 2; i <= sqrt(a); i++)
		if (a % i == 0)
			return false;
	return true;
}
void dfs(int m, int sum, int cnt)//第m个数,目前的和,选了几个数了已经
{
	if (m > n)
	{
		return;
	}
	if (cnt == k)
	{
		if (isprime(sum))
		{
			ans++;
		}
		return;
	}
	for (int i = m; i < n; i++)//下一个数从第m个数后面开始找
	{
		dfs(i + 1, sum + a[i], cnt + 1);
	}
	return;
}
int main()
{
	cin >> n >> k;
	for (int i = 0; i < n; i++)
	{
		cin >> a[i];
	}
	dfs(0, 0, 0);
	cout << ans << endl;
	return 0;
}