DFS-选数

Posted 桃花涣小鱼干

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了DFS-选数相关的知识,希望对你有一定的参考价值。

选数

问题来源:P1036 [NOIP2002 普及组] 选数

完整代码:

#include<math.h>
#include<stdio.h>
#include<stdlib.h>
int* p, ans = 0;
int isPrime(int m)

	int isPrime = 1;
	int k = sqrt(m);
	for (int i = 2; i <= k; i++)
	
		if (m % i == 0)
		
			isPrime = 0;
			break;
		
	
	if (isPrime == 0)
		return 0;
	else
		return 1;


//从x单元开始的选数
//已选m个数
//m个数的和为sum
//需要得到k个数,一共n个数
void dfs(int x,int m,int sum,int k,int n)

	if (m == k)
	
		if (isPrime(sum))
			ans++;
		return;
	
	for (int i = x; i < n; i++)
		dfs(i+1,m + 1, sum + *(p + i), k, n);//i+1而非x+1
	return;

int main()

	int n, k;
	scanf("%d%d", &n,&k);
	p = (int*)malloc(sizeof(int) * n);
	for (int i = 0; i < n; i++)
		scanf("%d", p + i);
	dfs(0,0,0,k,n);
	printf("%d",ans);

代码解析:

用DFS进行高级的枚举操作时,显然当枚举了K个数时,我们需要判断K个数的和是否为素数,其代码如下:

if (m == k)
	
		if (isPrime(sum))
			ans++;
		return;
	

枚举时我们用一个for循环来选选取当下DFS函数需要选入的数
代码如下:

for (int i = x; i < n; i++)
	dfs(i+1,m + 1, sum + *(p + i), k, n);//i+1而非x+1

问:为何需要一个选数的起始位点这个参数,即x?

答:每一层DFS函数选取一个数,选取哪一个数是由for循环里面的 i 决定的
假如是5个数里面选3个数,我们希望的是选第一个数然后再选两个,完成之后,判断其和是否为素数,再把最后一个数换成其下一位的数。
eg: 1 2 3 4 5
第一次选:
1 2 3
然后:
1 2 4
再然后:
1 2 5
我们选数的基准点是第一个数,每次扩选实际上都是选的i+1的那个数,起始位点不变,一直变的是相对位点,所以是i+1这个相对位点,而非x+1,如下:

for (int i = x; i < n; i++)
	dfs(i+1,m + 1, sum + *(p + i), k, n);//i+1而非x+1

END

以上是关于DFS-选数的主要内容,如果未能解决你的问题,请参考以下文章

DFS-选数

DFS-选数

dfs之选数问题

洛谷 P1036 选数背包型DFS/选or不选

P1036选数

P1036 选数