求助// C++ // DFS//第一个测试点过不了
查看原帖
求助// C++ // DFS//第一个测试点过不了
206691
zihy0119楼主2021/10/18 22:14

求助// C++ // DFS//第一个测试点过不了

以下是本蒟蒻代码,求助QAQ

#include <bits/stdc++.h>
using namespace std;
int n,k  ,ans=0;
int a[25];
bool prime(int x)
{
	if(x%2==0) return false;
	for(int i=2;i<=sqrt(x);i++)
	   if(x%i==0) return false;
	return true;
}
void dfs(int start,int cnt,int sum)
{
	//start开始选数
	if(cnt==k && prime( sum)==true) ans++;
	for(int i=start ;i<=n;i++)
	    dfs( i+1, cnt+1, sum+a[i]);
}
int main()
{
	scanf("%d %d",&n,&k);
	for(int i=1;i<=n;i++)
	   scanf("%d",&a[i]);
	dfs(1,0,0);
	cout<<ans;
	return 0;
}
2021/10/18 22:14
加载中...