P1036 [NOIP2002 普及组] 选数 题解

这把不炫技了，正经玩一把。

先看题：

# [NOIP2002 普及组] 选数

## 题目描述

已知 $n$ 个整数 $x_1,x_2,\cdots,x_n$，以及 $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 \le n \le 20$，$k<n$）。

第二行 $n$ 个整数，分别为 $x_1,x_2,\cdots,x_n$（$1 \le x_i \le 5\times 10^6$）。

## 输出格式

输出一个整数，表示种类数。

## 样例 #1

### 样例输入 #1

```
4 3
3 7 12 19
```

### 样例输出 #1

```
1
```

## 提示

**【题目来源】**

NOIP 2002 普及组第二题

 读完题目很显然 要从两个方面入手

1：列出数字 2：判断素数

一  列出数字

列出数字很明显是个递归，跟我的上一篇博客写法非常相像https://blog.csdn.net/Peterkouki/article/details/135050844?spm=1001.2014.3001.5501

void dfs(int x,int sum,int num) //  void 类型，用 x, sum, num 作为参数
{
    if(num==k) // 当选了 k 个数时
    {
        if(isPrime(sum)) cnt++; // 如果和为素数，计数加一
        return; // 返回上一层
    }
    for(int i=x;i<=n;i++) // 从 x 开始遍历
    {
        dfs(i+1,sum+a[i],num+1); // 选 a[i]，更新参数
    }
}

 二 判断素数

bool isPrime(int x)
{
    int i;
    for(i=2;i*i<=x;i++)  //优化算法 不用多说
    {
        if(x%i==0) return false;  //在1和其本身之外找到其他因数则不是素数
    }
    return true;
}

关键点处理完成 上代码

#include<bits/stdc++.h>
using namespace std;
int n,a[25],k,cnt=0;

bool isPrime(int x)
{
    int i;
    for(i=2;i*i<=x;i++)
    {
        if(x%i==0) return false;
    }
    return true;
}

void dfs(int x,int sum,int num) // void 类型，用 x, sum, num 作为参数
{
    if(num==k) // 当选了 k 个数时
    {
        if(isPrime(sum)) cnt++; // 如果和为素数，计数加一
        return; // 返回上一层
    }
    for(int i=x;i<=n;i++) // 从 x 开始遍历
    {
        dfs(i+1,sum+a[i],num+1); // 选 a[i]，更新参数
    }
}
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++) // 输入 n 个数
    {
        cin>>a[i];
    }
    dfs(1,0,0); // 从第一个数开始，初始和为 0，初始个数为 0
    cout<<cnt; // 输出结果
    return 0;
}