思路:
假定数字的排列是有序的,先选出k个数字,可以参照全排列的方法先将数字选出来。然后计算数字的和再去判断是否是素数。但实际上数字的排列是无序的,所以可以根据排列组合的公式算出倍数,相除即可得到正解。
#include <cstdio>
#include <iostream>
using namespace std;
long long n,k,sum,cnt,times,flag,a[21],vis[21];
void check(long long sum)
{
flag=0;//clear
if(sum!=2)
{
for(int i=2;i*i<=sum;i++)
{
if(sum%i==0)
{
flag=1;
break;
}
}
}
if(sum==2) flag=0;
if(flag==0) cnt++;
}
void dfs(long long depth)
{
if(depth==k+1)
{
//cout<<sum<<endl;
check(sum);//is_a_prime?
return;
}
for(int i=1;i<=n;i++)
{
if(vis[i]!=1)
{
vis[i]=1;
sum+=a[i];
dfs(depth+1);
sum-=a[i];
vis[i]=0;
}
}
}
int main()
{
cin>>n>>k;
for(int i=1;i<=n;i++) cin>>a[i];
dfs(1);
times=1;
for(int i=1;i<=k;i++) times*=i;
cout<<cnt/times<<endl;;
return 0;
}