2021.5.23
选数(洛谷)P1036 [NOIP2002 普及组] 选数
题目:输入n,k代表输入n个数,同时从n个数中选k个数相加,最后计算素数的个数
解题思路:递归DFS三个参数,n个数,计数k1,和相加sum,分情况讨论:选或不选,到最后一层判断计数,再判断素数
#include<bits/stdc++.h>
using namespace std;
int n,k;
int x[21];
int ans;
int isp (int a)
{
int flag=0;
if(a==2)return 1;
for(int i=2;i<=sqrt(a);i++)
{
if(a%i==0)return 0;
}
return 1;
}
void dfs(int st , int c , int sum)
{
if (st == n + 1){
if (c != k) return ;
if (isp(sum)) ans++;
return ;
}
// 选
dfs(st + 1 , c + 1 , sum + x[st]);
// 不选
dfs(st + 1 , c , sum);
}
int main()
{
cin>>n>>k;
for(int i=1;i<=n;i++)
cin>>x[i];
dfs(1,0,0);
cout<<ans<<endl;
return 0;
}