当前位置: 代码迷 >> 综合 >> codeup 问题 C: 【递归入门】组合+判断素数
  详细解决方案

codeup 问题 C: 【递归入门】组合+判断素数

热度:15   发布时间:2024-01-25 01:55:30.0

题目描述

已知 n 个整数b1,b2,…,bn

以及一个整数 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

常规的dfs遍历题,但有一点想了好久才想到,需要注意的是:选中数字后,如 3, 7, 19。即3+7+19只能算是一种和为素数的方案 。之前一直不能理解这一点,害的我样例输出一直是6!!!

C++代码如下:

#include<iostream>
#include<cmath>
#include<algorithm>
#include<map>
using namespace std;
const int maxn = 25;
bool vis[maxn] = {false};
int n, k, sum, cnt, arr[maxn];//检查数字num是否素数 
bool check(int num){if(num == 1)return false;for(int i = 2; i <= (int)sqrt(1.0 * num); i++)if(num % i == 0)return false;return true;
}
//step:目前在第step个位置尝试放入数字; pre:从第pre个数字开始放入(前面的数字不管,使选中的k个数字是由小到大排序) 
void dfs(int step, int pre){if(step == k + 1){if(check(sum))cnt++;return;}for(int i = pre; i <= n; i++){if(!vis[i]){//选中第i个数作为第step个位置 vis[i] = true;sum += arr[i];dfs(step + 1, i+1);//回溯vis[i] = false;sum -= arr[i]; }}return;
}int main(){scanf("%d%d", &n, &k);for(int i = 1; i <= n; i++)scanf("%d", &arr[i]);sum = 0;	//和 cnt = 0;	//和为素数的方案次数 dfs(1, 1);printf("%d\n", cnt);return 0;
}