P1036 选数

2019年12月1日 0 作者 Ender

CSP2019带失败之后做的第一道题,还是在老师办公室。新学校并没有竞赛,只能将就下了。

题目是素数判定和dfs生成全组合,啥?你问为啥做这么水的题?因为等会要颓废不想动脑子。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>

int n, k;
int st[1000];
bool vis[1000];
int ans = 0;
inline bool judge(int a) {
    for(int i = 2; i * i <= a; i++) {
        if(a  == 0) return false;
    }
    return true;
}

inline void dfs(int depth, int sum, int startX) {
    if (depth == k && judge(sum)) {
        ans++;
        return;
    }
    for(int i = startX; i < n; i++) {
        if(vis[i]) continue;
        vis[i] = true;
        dfs(depth + 1, sum + st[i], i + 1);
        vis[i] = false;
    }
}

int main() {
    std::cin >> n >> k;
    for (int i = 0; i < n; i++) std::cin >> st[i];
    dfs(0, 0, 0);
    std::cout << ans;
    return 0;
}

这里没什么难度,唯一比较有意思的地方在如何去重,有兴趣可以研究下不降原则。