洛谷 p1036. 选数 组合型递归
原题链接
简单
作者:
咸鱼堆上的猫
,
2024-04-12 15:50:10
,
所有人可见
,
阅读 1
#include<bits/stdc++.h>
using namespace std;
const int N =30;
int n,k;
int a[N],b[N];
int res;
bool is_prime(int x)
{
if(x==1) return false;
for(int i=2;i<=x/i;i++)
if(x%i==0)
return false;
return true;
}
void dfs(int u,int start)
{
if(u>k)
{
int sum=0;
for(int i=1;i<=k;i++)
{
sum = sum+b[i];
}
if(is_prime(sum)) res++;
return ;
}
for(int i=start;i<=n;i++)
{
b[u]=a[i];
dfs(u+1,i+1);
b[u]=0;//每次都把U写成i要注意
}
}
int main(){
cin>>n>>k;
for(int i=1;i<=n;i++)
cin>>a[i];
dfs(1,1);
cout<<res;
}