[2002年NOIP普及组] 选数

已知 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)
x1,x2,…,xn (1<=xi<=5000000)
输出
屏幕输出,格式为:
一个整数(满足条件的种数)。
样例输入
4 3
3 7 12 19
样例输出
1

数据范围小直接dfs爆搜

复制代码
#include<bits/stdc++.h>

using namespace std;

int num[100],vis[100],ans,sum,n,k;

bool prime(int x)//素数判断
{
    if(x<=2)
    {
        return true;
    }
    for(int i=2;i<=sqrt(x);++i)
    {
        if(x%i==0)
        {
            return false;
        } 
    }
    return true;
}

void DFS(int s,int x,int y)//s:上次相加的数的位置  x:现在的步数  y:到目前相加所得的数
{
    if(x==k+1)
    {
        if(prime(y))
        {
            sum++; 
        }
        return;
    }
    for(int i=s;i<=n;++i)
    {
        if(vis[i]==0)
        {
            vis[i]=1;//标记
            DFS(i,x+1,y+num[i]);//递归
            vis[i]=0;//回溯
        }
    }
}

int main()
{
    cin>>n>>k;
    
    for(int i=1;i<=n;++i)
    {
        cin>>num[i];
    }
    
    sort(num+1,num+n+1);
    
    DFS(1,1,0);
    
    cout<<sum;
    
    return 0;
} 
复制代码

 

posted @   Cirpt  阅读(121)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示