递归+素数判断

最新推荐文章于 2023-12-30 16:56:33 发布

原创最新推荐文章于 2023-12-30 16:56:33 发布 · 1.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#dfs #水题

dfs 专栏收录该内容

17 篇文章

订阅专栏

博客给出一个算法题，已知n个整数和整数k，需从n个整数中选k个相加，计算和为素数的组合有多少种。还给出了输入输出格式及样例，最后提到需要代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述
【题意】
已知 n 个整数 x1,x2,…,xn，以及一个整数 k（k＜n）。
从 n 个整数中任选 k 个整数相加，可分别得到一系列的和。
例如当 n=4，k＝3，4 个整数分别为 3，7，12，19 时，可得全部的组合与它们的和为：
3＋7＋12=22 3＋7＋19＝29 7＋12＋19＝38 3＋12＋19＝34。
现在，要求你计算出和为素数共有多少种。
例如上例，只有一种的和为素数：3＋7＋19＝29。
【输入格式】
第一行两个整数：n , k （1<=n<=20，k＜n）
第二行n个整数：x1,x2，…,xn （1<=xi<=5000000）
【输出格式】
一个整数（满足条件的方案数）。
【样例输入】
4 3
3 7 12 19
【样例输出】
1
嗯哼，代码。

#include<iostream>
using namespace std;
int n,r,a[110],v[110],b[110],ans;
int jg(int x){
    if(x<2)return 0;
    int t=sqrt(x+1);
    for(int i=2;i<=t;i++)if(x%i==0)return 0;
    return 1;
}
void dfs(int k){
    if(k==r+1){
        int s=0;
        for(int i=1;i<=r;i++)s+=b[a[i]];
        if(jg(s)==1)ans++;
    }
    else for(int i=a[k-1]+1;i<=n;i++)
            if(v[i]==1){
                v[i]=0;a[k]=i;
                dfs(k+1);a[k]=0;v[i]=1;
            }
}
int main(){
    ios::sync_with_stdio(false);cin.tie(0);
    cin>>n>>r;
    for(int i=1;i<=n;i++)cin>>b[i],v[i]=1;
    ans=0;a[0]=0;dfs(1);
    cout<<ans<<endl;
}