做题笔记 递归模拟（洛谷 P1036 选数）

题目传送门 P1036 选数

不想点进链接里的话看这里

思路分析

我们来分析一下题目
题目要求是在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
简单画了个图示

我们让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<n;i++){    //从第startx个开始取数
		if(v[i])      //判断有没有取过
			continue;
		v[i]=1; 
		dg(m+1,sum+a[i],i+1);    //递归，取数个数＋1，和加上，下一次从startx+1个数开始取
		v[i]=0; 
	}
	return ;  //ans,a[i],v[i]均为全局变量，所以不用返回值
}

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

#include<iostream>
#include<cstdio>
#include<cmath>
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;i++){
		if(v[i]) 
			continue;
		v[i]=1;
		dg(m+1,sum+a[i],i+1);
		v[i]=0;
	}
	return ;
}
int main(){
	cin>>n>>k;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	dg(0,0,0);   //这里是调用递归函数，因为开始递归时取数个数，和，起始值都是0，所以写dg（0,0,0）；
	cout<<ans;
	return 0;
}

END

PS：有参考洛谷上的题解