n表示输入n个数,k表示从n个数里选择k个数进行相加,输出最后求和结果是素数的个数。
#include <stdio.h>
#include<math.h>
int ans = 0, n, k, a[20];
int prime(int x) {
int i,flag=1;
for (i = 2; i <= (int)(sqrt(x)); i++) {
if (x % i == 0) {
flag=0;
}
}
return flag;
}
void rec(int start, int count, int sum) {
int i;
if (count == k && prime(sum)) {
ans++;
}
for (i = start; i <= n; i++) {
rec(i + 1, count + 1, sum + a[i]);
}
}
int main() {
int i;
scanf("%d %d", &n, &k);
for (i = 1; i <= n; i++) {
scanf("%d", &a[i]);
}
rec(1, 0, 0);
printf("%d", ans);
return 0;
}
接下来进行核心代码块的讲解
void rec(int start, int count, int sum) {
int i;
if (count == k && prime(sum)) {
ans++;
}
for (i = start; i <= n; i++) {
rec(i + 1, count + 1, sum + a[i]);
}
}
start表示数组的下标,conut用来记录已经选了的数,sum表示求和。如果已经数到了第k个,并且和是素数,ans就自增,ans表示最后符合条件的个数。然后就是for循环进行下一次判断,下一次的数组下标就是当前的数组下标+1,count也是逐个计数,sum每次加的是a[i]。
我当时在写的时候遇到的问题就是,没有传start,for循环每次都是从1开始的,这样就会出现一个数字被加了多次,这道题不能重复添加,所以在做这种题的时候,不能思维惯性,每次都是从1开始,要看题目要求能不能重复出现。