P1036 [NOIP2002 普及组] 选数

一只蓝色小鲨鱼

已于 2024-04-05 19:34:43 修改

阅读量414

点赞数 4

文章标签： c++

于 2024-04-05 19:23:38 首次发布

本文链接：https://blog.youkuaiyun.com/m0_64267361/article/details/137407125

版权

本文介绍了如何使用C++语言实现一个解决NOIP2002普及组中选数问题的算法，通过深度优先搜索（DFS）遍历并检查组合是否为素数。代码展示了如何处理输入数组，判断组合的素数性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[NOIP2002 普及组] 选数 - 洛谷

#include <bits/stdc++.h>
#define int long long //(有超时风险)
#define PII pair<int,int>
#define endl '\n'

using namespace std;

const int N=2e5+10,M=1e3+10;

int ans=0;
int a[N];
int n,k;

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

void dfs(int cnt,int sum,int st)
{
    if(cnt==k)
    {
        if(is_prime(sum))
            ans++;
        return ;
    }

    for(int i=st;i<=n;i++)
    {
        dfs(cnt+1,sum+a[i],i+1);
    }
    return;
}

signed main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);

    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    //为什么要规定顺序：因为是组合型枚举，比如题目样例3 + 7 + 19满足，那么 7 + 3 + 19就不行。
    //cnt,sum,st
    dfs(0,0,1);

    cout<<ans<<endl;

    return 0;
}