dfs简单例题 组合+判断素数

问题 C: 【递归入门】组合+判断素数
[命题人 : 外部导入]
时间限制 : 1.000 sec 内存限制 : 128 MB

题目描述

已知 n 个整数b1,b2,…,bn

以及一个整数 k（k＜n）。

从 n 个整数中任选 k 个整数相加，可分别得到一系列的和。

例如当 n=4，k＝3，4 个整数分别为 3，7，12，19 时，可得全部的组合与它们的和为：
　　　　3＋7＋12=22　　3＋7＋19＝29　　7＋12＋19＝38　　3＋12＋19＝34。
　　现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3＋7＋19＝29。

输入

第一行两个整数：n , k （1<=n<=20，k＜n）
第二行n个整数：x1,x2，…,xn （1<=xi<=5000000）

输出

一个整数（满足条件的方案数）。

样例输入 Copy

4 3
3 7 12 19
样例输出 Copy

1
一、AC代码

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<vector>
#include<cmath>
#include<algorithm>
using namespace std;
	int fac[20+5];
	int n, k, ans;
	int pl[20+5];
	int visited[20+5];
//观察素数表可知，素数除了2和3之外都在6的倍数左右
int is_prime(int x){
    if( x==2 || x==3 )
        return 1;
    //不在6的倍数左右的数字一定不是素数
    if( x%6!=1 && x%6!=5 )
        return 0;
    //如果x不是素数，那么小于x的平方根的数中一定有x的约数。如果没有那么x为素数
    double sqrtx = (int)sqrt(x*1.0);
    //在6的倍数左右的数字不一定是素数
    for( int i=5; i<=sqrtx; i+=6 )
        if( x%i==0 || x%(i+2)==0 )
            return 0;
 
    //其余即为素数
    return 1;
}
 
void init_vi()
{
    for(int i=0; i<n; i++)
        visited[i] = 0;
}
 
void dfs(int index, int sum, int count)  //index代表要取的数字在fac[]数组中的下标
{
    if(count==k)
    {
        if(is_prime(sum))
            ans++;
        return;
    }
    else if(count>k)
        return;
 
    for( int i=index; i<n; i++ )
    {
        if(!visited[i])
        {
            visited[i] = 1;
            dfs(i, sum+fac[i], count+1);
 
            visited[i] = 0;
        }
    }
}
 
int main(){
    int t;
    while(cin>>n>>k)
    {
        ans = 0; //记录方案数
        init_vi();
        for( int i=0; i<n; i++ )
        {
            cin>>t;
            fac[i] = t;
        }
        sort(fac,fac+n);
        dfs(0,0,0);
        cout<<ans<<"\n";
    }
}

二、思路
首先这里有几个地方如果使用好的方式可以极大缩短计算时间。

1. 判断素数

   观察素数表可知，素数都是在6的倍数左右的数字。

2. 使用快排首先对输入的数字进行升序排序，这样能够在dfs()省很多时间

3. 因为数字是从小到大排，所以dfs()中的for循环从当前数字往后到尾组合相加就可以了