acm 洛谷 P1036 选数题解【DFS】

最新推荐文章于 2020-02-06 22:08:45 发布

原创最新推荐文章于 2020-02-06 22:08:45 发布 · 307 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

39 篇文章

订阅专栏

博客介绍了一道DFS类型题目的解题思路。给出题目链接，思路是先对数组排序，再从左往右选K个数，确保不重复选择，还提及了代码但未给出具体内容。

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

题目类型：DFS
题目链接：https://www.luogu.org/problemnew/solution/P1036

思路：先将数组进行排序，然后依次从左往右选出K个即可，这样可以保证k个数不被重复选择

代码：

#include <iostream>
#include <algorithm>
#include <cmath>
#include <set>
using namespace std;

int N,K;
int arr[20];
int total = 0;
void DFS(int coun,int cur,int lastindex);
bool ifprime(int n);
int main(){
    cin>>N>>K;
    for(int i = 0;i<N;i++){
        cin>>arr[i];
    }
    sort(arr,arr+N);
    for(int i = 0;i<N-K+1;i++){
        DFS(1,arr[i],i);
    }
    cout<<total;
    return 0;
}

void DFS(int coun,int cur,int lastindex){
    if(coun == K &&ifprime(cur) ){
            total++;
    }
    for(int i = lastindex+1;i<N;i++)
        DFS(coun+1,cur+arr[i],i);
}

bool ifprime(int n){
    if(n == 1)
        return false;
    if (n == 2)
        return true;
    for(int i = 2;i<=int(sqrt(n));i++){
        if(n%i == 0){
            return false;
        }
    }
    return true;
}