[DFS] P1036 选数 (普及-)

Date: 2019/10/11
Degree of difficulty：Universal
Original question：P1036 选数

这道题我用了多个变量来递归
t是递归位置
sum是和
st是起始位置
因为前一个数肯定要比后一个大
所以dfs 中st就是 i+1

AC code【Plan 1】

//Author:PhilFan;
#include<bits/stdc++.h>
using namespace std;
int a[25],n,k,sum,cnt,t,c[25],b[25];
bool prime(int x){
	for(int i = 2; i <= sqrt(x);i++){
		if(x%i==0)	return 0;
	}
	return 1;
}
void dfs(int t,int sum,int st){//位置、和、起始位置
	if(t == k){
		if(prime(sum))	cnt++;return;
	}
	for(int i = st; i <= n; i++){//i的取值需要特别注意
		dfs(t+1,sum+a[i],i+1);	
	}
}
int main()
{
	scanf("%d %d",&n,&k);
	for(int i = 1; i <= n; i++){
		scanf("%d ",&a[i]);
	}
	dfs(0,0,1);//起始位置是1；
	printf("%d ",cnt);
	return 0;
}