这道题其实就是递归题中组合类的题目,也就是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;
}