P1036 [NOIP2002 普及组] 选数（DFS/数学）

 题目传送门

题目大意：

给定n个数字，得到从中取出k个数的和为素数的有多少个 

第一思路：

考虑使用DFS ,使用一个数组来存储k个位置，模拟作为一个加法式子的k个位置，然后对这k个位置使用DFS，因为每个位置都可以存放n个数字中的每一个数（注意使用过的数字不能再使用），递归终点就是k个位置全部填满的时候进行叠加判断是否为素数，如果是则计数，否则回溯

注意下面的代码没有实现去重操作，即会出现等式左边的数字全排列的情况

我最后直接ans除以k的全排列数字以去重

更正确的做法可看P1036 [NOIP2002 普及组] 选数 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include<iostream>
#include<vector>
#include<cmath>
#define int long long

using namespace std;

int n, k;
vector<int>nums;//存储可用数字
vector<bool>visited(25);//用来记录某一个数字是否已经使用过
vector<int>formual;//存储式子
int sum = 0, ans = 0;

bool isPrime(int num) {//判断num是不是素数，是则返回true，反之返回false
	int n = sqrt(num);
	for (int i = 2; i <= n; i++) {
		if (num % i == 0) {
			return false;
		}
	}
	return true;
}


void dfs(int index) {
	if (index == k) {
		sum = 0;
		for (int i = 0; i < k; i++) {
			sum += formual[i];
		}
		if (isPrime(sum)) {
			ans++;
		}
	}
	else {
		//在k个位置里
		for (int i = 0; i < n; i++) {
			if (!visited[i]) {//如果下标为i的数字还未使用
				visited[i] = 1;
				formual.push_back(nums[i]);
				dfs(index+1);
				visited[i] = 0;
				formual.pop_back();
			}
		}

	}
}

signed main()
{
	int value;
	cin >> n >> k;
	for (int i = 0; i < n; i++) {
		cin >> value;
		nums.push_back(value);
	}
	dfs(0);
	value = 1;
	for (int i = k; i >= 2; i--) {
		value *= i;
	}
	cout << ans / value << endl;;
	return 0;
}

 第二思路

根据题目意思，就是取出所有的组合情况（Cnk）的值判断是否为素数

因此我们直接实现全组合数的求解，最后再判断它们里面的素数个数有多少即可

        而实现全组合数的递归求解过程如下：

首先在一个长度为n的数组里取出k个数字，然后在一个长度为n-1的数组里取出k-1个数字，直到取够数量

考虑从右向左开始取数（不从左往右是因为那样要把取数后剩下的数组单独拿出来，但从右往左取直接用原数组就可以了）

#include<iostream>
#include<vector>
#include<cmath>
#define int long long
using namespace std;

int n, k;
vector<int>arr;
vector<int>sums;

bool isPrime(int num) {//判断num是不是素数，是则返回true，反之返回false
	int n = sqrt(num);
	for (int i = 2; i <= n; i++) {
		if (num % i == 0) {
			return false;
		}
	}
	return true;
}

void fun(int n, int k, int sum) {//求Cnk排列数
    if (k == 0) {
        sums.push_back(sum);//得到某一种组合之和
        return;
    } 
    for (int i = n; i >= k; i--) {
        fun(i-1,k-1, sum + arr[i]);
    }
}

signed main()
{
    int value,len,ans=0;
    cin >> n >> k;
    arr.push_back(0);
    for (int i = 0; i < n; i++) {
        cin >> value;
        arr.push_back(value);
    }
    fun(n,k ,0);
    len = sums.size();
    for (int i = 0; i < len; i++) {
        if (isPrime(sums[i])) {
            ans++;
        }
    }
    cout << ans << endl;
    return 0;
}