ZJNU 1888

 

#include<stdio.h>
#include<math.h>
__int64 a[30];
int num=0,cnt=0,n,k;
int prim(__int64 a)
{
int i;
if(a==1) return 0;
else if(a==2) return 1;
else for(i=2;i<=sqrt(a);i++)
if(a%i==0) return 0;
return 1;
}
void dfs(int x,__int64 sum,int cnt)
{
int i;
if(cnt==k)
{
if(prim(sum)) num++;
return;
}
for(i=x+1;i<=n;i++)
dfs(i,sum+a[i],cnt+1);
return ;
}
int main()
{
int t,i;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&k);
for(i=1;i<=n;i++)
scanf("%I64d",&a[i]);
num=0;cnt=0;
dfs(0,0,0);
printf("%d\n",num);
}
return 0;
}
/*
3
5 3
1 1 1 1 1
9
20 10
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
*/

标程

#include <stdlib.h>
#include<stdio.h>
#include <time.h>
#include <math.h>
long n, k, a[21], used[21] = {0}, ans = 0;
int prime(long x) {
long i;
if (x == 1) return 0;
else if (x == 2) return 1;
else {
for (i = 2; i <= sqrt(x); i++)
if (x % i == 0)
return 0;
return 1;
}
}
void dfs(long kk, long ss, long sum) {
long i;
if (kk == k) {
if (prime(sum) == 1)
ans++;
} else {
for (i = ss + 1; i <= n; i++)
if (!used[i]) {//这里的标记多余
used[i] = 1;
dfs(kk + 1, i, sum + a[i]);
used[i] = 0;
}
}
}
void write() {
printf("%d\n", ans);
}
int main() {
long i;
int tt;
scanf("%d",&tt);
while (tt--) {
scanf("%ld%ld",&n,&k);
ans = 0;
for (i = 1; i <= n; i++)
scanf("%ld", &a[i]);
dfs(0, 0, 0);
write();
}
return 0;
}