洛谷 1036.选数

这道题其实就是递归题中组合类的题目，也就是dfs的一种而已。

思路：我们知道，从4个数里面选出来3个，这就是典型的组合数问题，关键就是怎么选的问题。

首先我们看样例是怎么解释的，样例中我们看到，这里的数字都是从小到大进行相加然后得出结果的，也就是说，我们选数的顺序可以按照升序的顺序进行选取。

那么，在前面的题解中我也写到了组合型递归的题目，大家可以看前面的题解，按照那个模板进行模仿，特别注意一下，我们不需要判断这个数的选取状态如何，我们只需要在dfs中加入一个参数start就能避免选数的重复问题，这是因为我们选取的顺序就是从小到大排序的，既然位置已经递增了，不可能会出现比m更大的数了。因此数组里面的数也就一定是升序的才行，所以注意运用排序。如果有读者会剪枝，也可以对if语句中的判断进行剪枝，这样可以大大的减少时间复杂度。

上代码：

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cmath> 
#include<vector>
#include<algorithm>
#include<stack>
#include<queue>
#include<sstream>
#include<map>
#include<limits.h>
#include<set>
#define MAX 50
#define _for(i,a,b) for(int i=a;i<(b);i++)
#define ALL(x) x.begin(),x.end()
using namespace std;
typedef long long LL;
int n, m, counts;
LL arr[MAX];
LL res[MAX];
int st[MAX];
LL sum;
bool is_sushu(LL n) {
    if (n == 2)
        return true;
    if (n == 1 || n == 0)
        return false;
    int flag = 0;
    for (LL i = 2; i < n; i++) {
        if (n % i == 0)
        {
            flag = 1;
            break;
        }
    }
    if (flag)
        return false;
    else
        return true;
}
void dfs(int u,int start) {
    if (u > m) {
        sum = 0;
        for (int i = 1; i <= m; i++) {
            sum += res[i];
        }
        if (is_sushu(sum))
            counts++;

        return;
    }
    for (int i = start; i <= n; i++) {
        res[u] = arr[i];
        dfs(u + 1, i + 1);
        res[u] = 0;
    }
}
int main() {
    
    ios::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    cin >> n >> m;
    _for(i, 1, n+1)
        cin >> arr[i];
    dfs(1, 1);
    cout << counts << endl;
    return 0;
}