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

猜你喜欢

转载自blog.csdn.net/weixin_51656756/article/details/120983177