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