CQOJ921B素数和

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

首先题意为给定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[];

int ans=;

bool judge(int x){

    if(x==) return true;

    for(int i=;i*i<=x;i++){

        if(x%i==) return false;

    }

    return true;

}

void dfs(int now,int step,int sum){//当前枚举到第几个，当前有几个数，当前的和

    if(step==k){

        if(judge(sum)) ans++;

        return;

    }

    for(int i=now+;i<=n;i++){

        dfs(i,step+,sum+a[i]);

    }

}

int main(){

    cin>>n>>k;

    for(int i=;i<=n;i++){

        cin>>a[i];

    }

    dfs(,,);

    cout<<ans;

    return ;

}

巴特西

CQOJ921B素数和

最新文章

热门文章