[NOIP2002]选数

题目来自牛客网（蓝桥杯集训第3期）

重点：<深度搜索DFS><回溯算法：组合问题>

题目描述

已知 n 个整数 x1,x2,…,xn，以及一个整数 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）
x1,x2，…,xn （1<=xi<=5000000）

输出描述:

输出格式为：一个整数（满足条件的种数）。 

示例1：

//输入

4 3
3 7 12 19

//输出

1

思路

1.判断是否是素数

求素数的方法：接收一个整数 x 作为参数，然后通过遍历2到sqrt(x)的范围，检查是否有除1和自身以外的因子，若有则返回 false，否则返回 true。

int prime(int x)  
{  
    for(int i=2;i<=sqrt(x);i++)  
    {  
        if(x%i==0)  
            return 0;  
    }  
            return x;  
}

2.深度搜索

（组合问题：回溯算法）（深度搜索DFS）

这是一个深度优先搜索（DFS）的递归函数。

它有三个参数：

sum表示当前子序列的和;

d表示当前子序列的长度;

t表示当前考虑的元素的下标。

如果d等于k（在主函数中输入的值），并且sum是一个质数，则将ans的值加1。否则，对于数组中的每一个元素（从下标t到n-1），递归调用DFS函数。

void DFS(int sum,int d,int t)
{
    if(d==k)
    {
        if(prime(sum))
            ans++;
    }
         
    else
    {
        for(int i=t;i<n;i++)
        {
            DFS(sum+arr[i],d+1,i+1);
        }
    }
}

3.主函数

int main()  
{  
    scanf("%d %d",&n,&k);  // 从标准输入读取两个整数n和k。  
    for(int i=0;i<n;i++)  
    {  
        scanf("%d",&arr[i]); // 从标准输入读取n个整数并存储在数组arr中。  
    }  
    DFS(0,0,0); // 调用DFS函数，开始搜索。搜索的起始条件是sum=0, d=0, t=0。  
    printf("%d",ans); // 打印出搜索结果。即所有长度为k且和为质数的子序列的数量。  
    return 0; // 主函数结束，返回0。  
}

完整代码

#include<stdio.h>
#include <math.h>
int n,k,ans;
int arr[1024];

int prime(int x)
{
    for(int i=2;i<=sqrt(x);i++)
    {
        if(x%i==0)
            return 0;
    }
            return x;
}

void DFS(int sum,int d,int t)
{
    if(d==k)
    {
        if(prime(sum))
            ans++;
    }
         
    else
    {
        for(int i=t;i<n;i++)
        {
            DFS(sum+arr[i],d+1,i+1);
        }
    }
}

int main()
{
    scanf("%d %d",&n,&k);  
    for(int i=0;i<n;i++)
    {
        scanf("%d",&arr[i]);
    }
    DFS(0,0,0);
    printf("%d",ans);
    return 0;
}