P1036 选数

弑魔者魔眼

1. #include <bits/stdc++.h>
   
2. using namespace std;
   
3. int a[30];
   
4. bool check(int x)
   
5. {
   
6.         for(int i=2;i*i<=x;i++)
   
7.         if(x%i==0) return 0;
   
8.         return 1;
   
9. }
   
10. 
    
11. int main()
    
12. {
    
13.         int n,k,ans=0;
    
14.         cin>>n>>k;
    
15.         for(int i=0;i<n;i++) cin>>a[i];
    
16.         int U=1<<n;
    
17.         for(int S=0;S<U;S++)
    
18.         if(__builtin_popcount(S)==k)
    
19.         {
    
20.                 int sum=0;
    
21.                 for(int i=0;i<n;i++)
    
22.                 if(S&(1<<i)) sum+=a[i];
    
23.                 if(check(sum)) ans++;
    
24.         }
    
25.         cout<<ans;
    
26.         return 0;
    
27. }

复制代码