中石油oj周赛：选取

【题目描述】
给定n个整数a[i]，你需要求出有多少个从中选出k个的方案，使得这k个整数的和是质数。
两种方案被视为不同，当且仅当存在一个i满足a[i]只在其中一种方案中被选出。

输入
第一行两个整数n和k，第二行n个整数a[i]。

输出
一行一个整数，表示从n个整数中选出k个数的和是质数的方案数。

输入样例

4 3
3 7 12 19

输出样例

1

提示
对于30%的数据，k=1；
对于另40%的数据，n<=5；
对于所有数据，1<=n<=20，k<n，1<=a[i]<=5000000。

思路1：二进制枚举
观察题目发现，n的数据量较小，并且一定满足k≤n，并且k的值不定，2^20在1e6的量级范围内，因此是一个合理的范围，所以可利用二进制进行状态压缩。
对于题目要求不同的方案，对数据进行去重即可。
思路1前置知识：二进制、lowbit、试除法判定质数

#include <iostream>
#include <cstring>
#include <map>
#include <algorithm>
using namespace std;

const int N = 25;
int n, k;
int q[N];
map<int, int> id_query;      //<1<<i, i>，打表查询二进制
int res = 0;

inline int lowbit(int x) { return x & -x; }

int check(int x) {     //检查x的二进制中1有多少个
    int cnt = 0;
    while(x) {
        cnt ++;
        x -= lowbit(x);
    }
    return cnt;
}

bool is_Prime(int x) {
    if(x < 2) return false;
    for(int i = 2; i <= x / i; i ++) {
        if(x % i == 0) return false;
    }
    return true;
}

int main() {
    //打表
    for(int i = 0; i <= 20; i ++) id_query[1 << i] = i;
    
    scanf("%d%d", &n, &k);
    for(int i = 0; i < n; i ++) scanf("%d", &q[i]);
    n = unique(q, q + n) - q;           //小技巧：n是去重后新数组的大小
    
    for(int i = 0; i < (1 << n); i ++) {
        if(check(i) == k) {
            int t = i;
            int tmp_sum = 0;
            while(t) {
                tmp_sum += q[id_query[lowbit(t)]];
                t -= lowbit(t);
            }
            if(is_Prime(tmp_sum)) res ++;
        }
    }
    
    cout << res << endl;
    
    return 0;
}

思路2：暴搜
一个类似的题目：https://www.acwing.com/problem/content/95/

暴搜的搜索顺序即为数组下标从小到大搜索。
代码：

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

#define lowbit(x) (x & -x)

const int N = 25;
int q[N], n, k;
int path[N];       //选取了多少个数

int res = 0;

bool is_Prime(int x) {
    if(x < 2) return false;
    for(int i = 2; i <= x / i; i ++) {
        if(x % i == 0) return false;
    }
    return true;
}

//优化搜索顺序：按照下标大小从小到大搜，选k个
void dfs(int u, int last) {
    if(u == k) {
        //for(int i = 0; i < k; i ++) cout << path[i] << ' ';
        //puts("");
        
        int tmp_sum = 0;
        for(int i = 0; i < k; i ++) tmp_sum += path[i];
        if(is_Prime(tmp_sum)) res ++;
        
        return;
    }
    
    for(int i = last + 1; i < n; i ++) {
        path[u] = q[i];
        dfs(u + 1, i);
        path[u] = 0;
    }
}

int main() {
    scanf("%d%d", &n, &k);
    for(int i = 0; i < n; i ++) scanf("%d", &q[i]);
    n = unique(q, q + n) - q;
    
    dfs(0, -1);
    
    cout << res << endl;
    
    return 0;
}