码迷,mamicode.com
首页 > 其他好文 > 详细

CQOJ921B素数和

时间:2019-09-21 21:14:45      阅读:95      评论:0      收藏:0      [点我收藏+]

标签:string   main   pac   过程   ati   记录   namespace   枚举   ace   

这是用一道搜索(全排列)实现的一个数论题目。今天周六,上了一天信息,没写数学的我瑟瑟发抖。

首先题意为给定n个数,选取k个数进行求和,输出和为素数的方案数。在写判断素数函数时,我们只需要把i枚举到根下x即可,这一点可以证明。在写全排列列时,由于我们需要枚举的是和,所以不需要再考虑顺序了,要用到三个变量(int now,int step,int sum){表示当前枚举到第几个,当前有几个数,当前的和,生成一种排列后直接判断即可。另lyx tql

1.牢记素数判断优化。

2.想搜索的过程中需要几个变量记录状态。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#define maxn 21
using namespace std;
int n,k;
int a[100];
int ans=0;
bool judge(int x){
    if(x==1) return true;
    for(int i=2;i*i<=x;i++){
        if(x%i==0) return false;
    }
    return true;
}
void dfs(int now,int step,int sum){//当前枚举到第几个,当前有几个数,当前的和 
    if(step==k){
        if(judge(sum)) ans++; 
        return;
    }
    for(int i=now+1;i<=n;i++){
        dfs(i,step+1,sum+a[i]);
    }
}
int main(){
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }    
    dfs(0,0,0);
    cout<<ans;
    return 0;
}

 

CQOJ921B素数和

标签:string   main   pac   过程   ati   记录   namespace   枚举   ace   

原文地址:https://www.cnblogs.com/china-mjr/p/11564297.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!