当前位置: 代码迷 >> 综合 >> 洛谷——P1036 [NOIP2002 普及组] 选数
  详细解决方案

洛谷——P1036 [NOIP2002 普及组] 选数

热度:98   发布时间:2023-12-05 20:01:15.0

题目描述

已知 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)。

第二行 n 个整数,分别为 x1?,x2?,?,xn?(1≤xi?≤5×1e6)。

输出格式

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

输入输出样例

输入 #1复制

4 3
3 7 12 19

输出 #1复制

1

说明/提示

【题目来源】

NOIP 2002 普及组第二题

解题方法:DFS.

Code:

#include <iostream>
#include <stdio.h>using namespace std;typedef int Status;const int N = 5000010;int n, k, res;
int a[N], p[N];
bool st[N];Status prime(int x)
{if(x == 0 || x == 1)  return false;bool flag = true;for(int i = 2; i * i <= x; i ++ )  if(x % i == 0)  flag = false;return flag;
}void dfs(int x, int y, int z)
{if(x == k){if(prime(y))  res ++;return ;}for(int i = z; i <= n; i ++ ){if(!st[i]){st[i] = true;dfs(x + 1, y + a[i], i + 1);st[i] = false;}}
}signed main()
{scanf("%d%d", &n, &k);for(int i = 1; i <= n; i ++ )  scanf("%d", &a[i]);dfs(0, 0, 1);cout << res << endl;
}