做题笔记 递归模拟(洛谷 P1036 选数)

题目传送门 P1036 选数

不想点进链接里的话看这里
做题笔记 递归模拟(洛谷 P1036 选数)_第1张图片

思路分析

我们来分析一下题目
题目要求是在n个数里面选k个相加等于素数,那么我们第一步就是要写好一个判断素数的函数
判断素数很简单,我之前在洛谷写过一个博客,大家可以去看看 判断素数最简单的三种方法,这里我就直接贴代码

//判断素数
bool isprime(int n){
	for(int l=2;l<=sqrt(n);l++){ //sqrt开方函数要加上头文件cmath
		if(n%l==0) return 0;
	}
	return 1;
}

OK我们接着往下写
n个数取k个,要求出都多少种取法,那当然是用递归,但是这里还有一个去重的问题
拿在4个数里取2个举例,假设四个数为1 2 3 4
简单画了个图示
做题笔记 递归模拟(洛谷 P1036 选数)_第2张图片
我们让1和2,3,4分别组合,再从2开始让2与大于2的数组合,最后是从3开始和4组合,到4这里就不能往后取了,结束
这样取数是小学就学过的技巧,可以有效去重
所以我们可以递归模拟这个取数的过程
上代码

//递归,代码的主要部分
void dg(int m,int sum,int startx){ //m为取了多少个数,sum是取好的数相加的和
//这个startx要注意,它是代表取数的起始值,也就是每次从第startx个开始
	if(m==k){     //如果选好了k个,判断和是不是素数,是的话计数器+1,不是就return
		if(isprime(sum))
		  	ans++;
		return ;
	}
	for(int i=startx;i

递归的函数也写好了,那么可以写主函数输入数据了
最后放上完整AC代码

#include
#include
#include
using namespace std;
bool isprime(int n){
	for(int l=2;l<=sqrt(n);l++){
		if(n%l==0) return 0;
	}
	return 1;
}
int n,k,a[25];
bool v[25];
long long ans=0;
void dg(int m,int sum,int startx){
	if(m==k){
		if(isprime(sum))
		  	ans++;
		return ;
	}
	for(int i=startx;i>n>>k;
	for(int i=0;i>a[i];
	}
	dg(0,0,0);   //这里是调用递归函数,因为开始递归时取数个数,和,起始值都是0,所以写dg(0,0,0);
	cout<

END

PS:有参考洛谷上的题解

你可能感兴趣的:(做题笔记)