洛谷 P1036 选数 题解 （深搜）C语言

    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开始，要看题目要求能不能重复出现。