cggwz  

题目链接:选数
这一题水过去就行了,我们这里用next_permutation去生成各种排列,有一个注意点,我会在代码中标注。

#include<bits/stdc++.h>
using namespace std;
bool isprime(int x){
    int q=sqrt(x+0.5);
    for(int i=2;i<=q;i++){
        if(x%i==0){
            return false;
        }
    }
    return true;
}
int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    int num[n];
    for(int i=0;i<n;i++){
        scanf("%d",&num[i]);
    }
    int vis[n];
    memset(vis,0,sizeof(vis));
    for(int i=0;i<k;i++){
        vis[n-1-i]=1;               //1
    }
    int ans=0;
    bool running=true;
    while(running){
        int sum=0;
        for(int i=0;i<n;i++){
            if(vis[i]==1){
                sum+=num[i];
            }
        }
        if(isprime(sum)){
            ans++;
        }
        running=next_permutation(vis,vis+n);
    }
    printf("%d",ans);
    return 0;
}

就一处:
1处:注意我的赋值,我将最后几项设置为1,这样才会产生全排列,因为这个函数,是按字典序生成的排列,如果赋值开头几个,循环一遍就结束,所以这一点很重要。

posted on 2017-08-15 21:51  cggwz  阅读(131)  评论(0)    收藏  举报