洛谷-P1036-选数

洛谷-P1036-选数

#include <bits/stdc++.h>
using namespace std;

int ans=0;
int n,k;
int sz[20];

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

void dfs(int start, int count, int sum) {
    //start：开始选数的地方
    //count：已经选了几个数
    //sum：到目前为止选的数的和
	if (count == k && isPrime(sum)) {
		ans++;
    	//如果已经选了k个数，判断和是不是素数
        //ans是符合条件的和的个数
	}

	for (int i = start; i <= n; i++) {
    	//从开始选数的地方到n
        //每重循环都是一种可能性
        //例如：选第二个数，start=2,n=4,有三种可能性，调用自身3次

        dfs(i + 1, count + 1, sum + sz[i]);

        //参数1：从a里面下一个数开始选
        //参数2：已经选的数的个数+1
        //参数3：sum加上这次选的数

        //因为直接从下一个数开始选，所以不可能选到之前选过的数
        //无需判断当前的数是否被选过
	}
}

int main() {

	int i;
	cin >> n >> k;
	for (i = 1; i <= n; i++) {
		cin >> sz[i];
	}
	dfs(1, 0, 0);
    //从第一个数开始找，已经找了0个数，目前的和是0
	cout << ans << endl;
	return 0;
}