17分dfs求助
查看原帖
17分dfs求助
1310268
Pocon楼主2024/3/25 20:52
#include<iostream>
using namespace std;

int n,m;   //n表示一共有多少数,m表示一共有多少位置
const int N=25;
int k[N];
int arr[N];
int count=0;

bool isprime(int n){
	if(n<2)  return false;
	for(int i=2;i*i<=n;i++){
		if(n%i==0)  return false;
	}
	return true;
}

void dfs(int x,int start){
	int sum=0;    
	if(x>m){
		for(int i=1;i<=m;i++){
			sum+=arr[i];	
		}
		if(isprime(sum))  count++;
		return ;
	}
	
	for(int i=start;i<=n;i++){
		arr[i]=k[i];
		dfs(x+1,i+1);
		arr[i]=0;
	}
}

int main(){
	cin >> n >> m;
	for(int i=1;i<=n;i++){
		cin >> k[i];
	}
	dfs(1,1);
	cout << count << endl;
	return 0;
}
2024/3/25 20:52
加载中...