蒟蒻全WA求调
查看原帖
蒟蒻全WA求调
798795
___Segment___楼主2024/7/23 10:13
#include<bits/stdc++.h>
using namespace std;
const int MAX=20;
int a[MAX+1],n,k,c,d[MAX+1];
bool vis[MAX+1];
bool isprime(int n)
{
	if(n<2) return 0;
	int t=sqrt(n);
	for(int i=2;i<=t;i++)
	{
		if(n%i==0) return 0;
	}
	return 1;
}
void dfs(int deep)
{
	if(deep==k+1)
	{
		int s=0;
		for(int i=1;i<=k;i++) s+=d[i];
		if(isprime(s)) c++;
		return;
	}
	for(int i=1;i<=n;i++)
	{
		if(!vis[i])
		{
			vis[i]=1;
			d[deep]=a[i];
			dfs(deep+1);
			vis[i]=0;
		}
	}
} 
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	dfs(1);
	printf("%d",c);
	return 0;
}
2024/7/23 10:13
加载中...