OJ 1381 选数

描述

已知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）。

第二行输入n个整数x1,x2,…,xn（1<=xi<=5000000）。

输出
对于每组输入数据，输出一个整数，表示满足条件的种数。

输入样例 1

4 3
3 7 12 19
输出样例 1

1

 

题目要求是从n个数中找出k个数的和，并且和为素数，那么可以使用dfs搜索的方式来计算剪枝，从第一个开始向后遍历，通过一节一节的递归找出每个位置的组合即可，当和为素数则计数。
 

#include <iostream>
#include <cmath>
using namespace std;
int a[25];
int n,k,value=0;
bool prime(int x)
{
    for(int i=2;i<=sqrt(x);i++)
        if(x%i==0)
         return false;
    return true;
}
void dfs(int sum,int location,int num)
{
    if(num==k)
    {
        if(prime(sum))
          value++;
     }
    else
    {
        for(int i=location;i<n;i++)
            dfs(sum+a[i],i+1,num+1);
    }
}

int main()
{
   while(cin>>n>>k)
   {
       value=0;
       for(int i=0;i<n;i++)
        cin>>a[i];
    dfs(0,0,0);
    cout<<value<<endl;
   }
   return 0;

}