蒟蒻初学dfs 样例已过 测得只有33分 实在不知道错哪了 求大神QAQ
查看原帖
蒟蒻初学dfs 样例已过 测得只有33分 实在不知道错哪了 求大神QAQ
472945
武怡雯_楼主2021/7/21 16:18
#include<iostream>
#include<cmath>
using namespace std;

long long n,x[50],k,ans;

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

void dfs(int z , int sum , int ks)
{
	if(z == k) {
		if(zs(sum)) ans++;
		return;
	}
	for(int i = ks; i < n; ++i) {
		dfs(z + 1 , sum + x[i] , i + 1);
	}
}

int main()
{
	cin>>n>>k;
	for(int i = 0; i < n; ++i) {
		cin>>x[i];
	}
	dfs(0 , 0 , 0);
	cout<<ans;
	return 0;
}
2021/7/21 16:18
加载中...