P1036 [NOIP2002 普及组] 选数

// Problem: P1036 [NOIP2002 普及组] 选数
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P1036
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// User: Pannnn

#include <bits/stdc++.h>

using namespace std;
/*
    一共n个数，每个数有两种状态：使用和不使用，使用二进制遍历
    一个二进制有n位的数，每一位可为0或1，从0开始遍历到(1 << n) - 1
    即可枚举出所有可能状态，第i位为0表示第i位不使用
*/
bool isPrim(int n) {
    for (int i = 2; i * i <= n; ++i) {
        if (n % i == 0) {
            return false;
        }
    }
    return true;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    
    int n, k;
    cin >> n >> k;
    vector<int> info(n);
    for (int i = 0; i < n; ++i) {
        cin >> info[i];
    }
    
    int res = 0;
    for (int i = 0; i < (1 << n); ++i) {
        int cnt = 0;
        int sum = 0;
        for (int j = 0; j < n; ++j) {
            if ((i & (1 << j)) != 0) {
                ++cnt;
                sum += info[j];
            }
        }
        if (cnt == k && isPrim(sum)) {
            ++res;
        }
    }
    cout << res << endl;
    return 0;
}

标签：info,NOIP2002,int,选数,P1036,Limit
来源： https://www.cnblogs.com/pannnn/p/15867906.html