选数(NOIP2002)

题目链接：选数

这一题水过去就行了，我们这里用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，这样才会产生全排列，因为这个函数，是按字典序生成的排列，如果赋值开头几个，循环一遍就结束，所以这一点很重要。