P1036 [NOIP2002 普及组] 选数

fxc1437936516

已于 2022-12-18 17:01:22 修改

阅读量58

点赞数

分类专栏：洛谷官方提单文章标签：算法 c++ 深度优先

于 2022-12-18 13:01:59 首次发布

本文链接：https://blog.csdn.net/fxc1437936516/article/details/128361394

版权

洛谷官方提单专栏收录该内容

6 篇文章 1 订阅

订阅专栏

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 30;

int n, k;
int q[N];
int ans;

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

void dfs(int cnt, int sum, int str)
{
    if(cnt == k) 
    {
        if(is_prime(sum)) ans ++;
        return ;
    }
    
    for(int i = str; i < n; i ++)
        dfs(cnt + 1, sum + q[i], i + 1);
}

int main ()
{
    cin >> n >> k;
    for(int i = 0; i < n; i ++) cin >> q[i];
    
    dfs(0, 0, 0);
    
    cout << ans << endl;
    
    return 0;
}