洛谷 P1036 选数

题目描述已知 n 个整数 x1,x2,…,xn，以及1个整数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个整数（满足条件的种数）。

输入输出样例
输入样例#1：
4 3
3 7 12 19
输出样例#1：
1

递归好难写啊…@_@脑仁疼
下面是洛谷的其中三种解法：

递归1：

#include<iostream>
#include<stdio.h>
#include<math.h>
using namespace std;
int n, k;
int a[20];
bool prime(int sum)//判断素数
{
 int sqr = sqrt((double)sum);
 for (int i = 2; i <= sqr; i++)
 {
  if (sum%i == 0)
   return false;
 }
 return true;
}
int anser(int left, int sum, int sta, int end)
//left是剩余的可选数的个数，sum就是k个数的和，sta和end是可选数的范围
{
 /*printf("left=%d,sum=%d,sta=%d\n ",left,sum,sta);*/
 if (left == 0)//没有数可选时判断是不是素数
  return prime(sum);
 int ans = 0;
 for(int i=sta;i<=end;i++)
 { 
  /*printf("i=%d\n",i);
  printf("left=%d,sum=%d,sta=%d\n ", left, sum, sta);
  printf("----------------------------------------------\n");*/
  ans+=anser(left-1,sum+a[i],i+1,end);
  //递归，可选数减一，sum加上a[i],可选范围后移一位
  //注意i的更新
 }
  
 return ans;
}
int main()
{
 scanf("%d%d", &n, &k);
 for (int i = 0; i < n; i++)
 {
  scanf("%d", &a[i]);//输入a[i]
 }
 
 printf("%d\n", anser(k,0,0,n-1));
 //system("pause");
 return 0;
}

递归2：

#include<iostream>
#include<stdio.h>
#include<math.h>
using namespace std;
int n, k, a[21],sum=0,ans=0;
bool b[21];//定义一个bool类型的数组记录该数有没有被加过
int prime(int sum)//判断素数函数
{
 int s = sqrt(sum);
 for (int i = 2; i <= s; i++)
 {
  if (sum%i == 0)
   return 0;
 }
 return 1;
}
int count(int t, int cnt)
//t用来计算加上了多少个数,cnt用来遍历1~n个数
{
 for (int i = cnt; i <= n; i++)
 {
  if (b[i] == true)//如果第i个数没有用过
  {
   b[i] = false;//使用
   sum += a[i];
   if (t == k)//当加上满k个数时，判断sum是不是素数，并更新答案
    ans+=prime(sum);
   else
    count(t + 1, i + 1);//否则递归下一轮
  
    b[i] = true;//回溯前要把该数更新为未用过的数，才能判断下一情况
    sum -= a[i];//sum也减去当前数的值
    //注意他俩在else之外，因为t==k时也需要回溯更新
  }
 }
}
int main()
{
 cin >> n >> k;
 for (int i = 1; i <= n; i++)
 {
  scanf("%d", &a[i]);
  b[i] = true;//初始化数组
 }
 count(1, 1);
 printf("%d\n",ans);
 //system("pause");
 return 0;
}

dfs：
（dfs感觉如此美妙）

#include<iostream>
#include<stdio.h>
#include<math.h>
using namespace std;
int n, k, a[21];
int ans=0;
int prime(int sum)
{
 if (sum == 1 || sum == 0)
  return 0;
 int s = sqrt(sum);
 for (int i = 2; i <= s; i++)
 {
  if (sum%i == 0)
   return 0;
 }
 return 1;
}
void dfs(int cnt,int sum,int t)
//用cnt来记录第i个数，sum为和，t为加上的数的个数
{
 if (cnt == n + 1 || t == k+1)//如果加到了第n个数，或者加满了k个数
 {
  if (prime(sum) && t == k+1)//如果是素数，并且加满了k个数
   ans++;//答案加一
  return;
 }
 dfs(cnt + 1, sum + a[cnt], t + 1);//sum加上当前数，继续dfs下一个数
 dfs(cnt + 1, sum, t);//跳过这个数不加，dfs下一个数
 return;
}
int main()
{
 cin >> n >> k;
 for (int i = 1; i <= n; i++)
 {
  scanf("%d", &a[i]);//输入数
 }
 dfs(1, 0, 1);//从第一个数开始加
 printf("%d\n", ans);
 //system("pause");
 return 0;
}