洛谷训练 P1036 选数

洛谷训练 P1036 选数

题目描述

已知 n 个整数 x1,x2,…,xn，以及1个整数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)

输出格式：

屏幕输出，格式为： 1个整数（满足条件的种数）。

输入输出样例

输入样例#1：
4 3
3 7 12 19
输出样例#1：
1

传送门

题目链接: https://www.luogu.org/problemnew/show/P1036.
素数的判定: https://blog.csdn.net/jonms/article/details/80464622.
从N个数选取k个数的组合: https://blog.csdn.net/lhc121386/article/details/86697977.

AC代码

#include <iostream>
#include <cstring>

/*
思路:用dfs和不降原则（避免重复组合）进行组合数，在选取组合数时和相加；用素数的定义来判断素数；
重点：1.从N个数中选取k个数的组合--不降原则（DFS）
      2.素数的判定
*/

using namespace std;
int n,k;
int a[25];
int vis[25];
int ans;

bool prime(int num)//用素数的定义判断素数
{
    for(int i=2;i*i<=num;i++)
    {
        if(num%i==0)
        {
            return false;
        }
    }
    return true;
}

void dfs(int step,int sum,int strat)
{
    if(step==k)
    {
        if(prime(sum))
        {
            ans++;
        }
        else
        {
            return;
        }
    }
    for(int i=strat;i<n;i++)//不降原则核心步骤1
    {
        if(vis[i]==0)
        {
            vis[i]=1;
            dfs(step+1,sum+a[i],i+1);//不降原则核心步骤2
            vis[i]=0;
        }
    }
    return;
}

int main()
{
    while(cin>>n>>k)
    {
        ans=0;
        memset(vis,0,sizeof(vis));
        memset(a,0,sizeof(a));
        for(int i=0;i<n;i++)
        {
            cin>>a[i];
        }
        dfs(0,0,0);
        cout<<ans<<endl;
    }
}