luogu 1036

简单的深搜加上判断是否是质数。

由于是组合，所以要考虑顺序，不然会超时。

#include"cstdio"
#include"cmath"
int n,k,ans=0,a[21];
int check(int w)
{
    if(w==1) return 0;
    int n=sqrt(w)+0.5;
    for(int i=2; i<=n; i++)
        if(w%i==0) return 0;
    return 1;
}
void dfs(int t,int l,int now)
{
    if(t==k+1)
    {
        if(check(now)) ++ans;
        return;
    }
    for(int i=l; i<=n-k+t; i++)
        dfs(t+1,i+1,now+a[i]);
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1; i<=n; i++) scanf("%d",&a[i]);
    dfs(1,1,0);
    printf("%d",ans);
    return 0;
}

其实可以用状压的方法来递推。

f[i]表示状态为i时数的和是多少。

#include"cstdio"
#include"cmath"
int n,k,ans,a[20],f[1<<20];
int calc(int x)
{
    int tot=0;
    for(; x; x>>=1) tot+=x&1;
    return tot;
}
int check(int w)
{
    if(w==1) return 0;
    int n=sqrt(w)+0.5;
    for(int i=2; i<=n; i++)
        if(w%i==0) return 0;
    return 1;
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=0; i<n; i++) scanf("%d",&a[i]);
    for(int i=0; i<(1<<n); i++)
    {
        if(calc(i)==k && check(f[i])) ans++;
        for(int j=0; j<n; j++)
            if((i&(1<<j))==0) f[i|(1<<j)]=f[i]+a[j];
    }
    printf("%d",ans);
    return 0;
}