P1036选数

#include<bits/stdc++.h>
using namespace std;
using i64 = long long;
using u64 = unsigned long long;
using u32 = unsigned;
int arr[22];
int n,k,ans;
bool isPrime(int a)
{
    for(int i=2; i*i <=a; i++)
    {
        if(a%i==0)
        {
            return 0;
        }
    }
    return 1;
}

void dfs(int i , int nums , int sum)
{
    if(nums == k)
    {
        if(isPrime(sum))
        {
            ans++;
        }
    }else if(i <= n)
    {
        dfs(i+1, nums+1, sum+arr[i]);
        dfs(i+1, nums, sum);
    }
}


int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);


    cin >> n >> k;

    for(int i = 1; i <= n; i++) cin >> arr[i];
    dfs(1,0,0);
    cout << ans;
    return 0;
}