【递归入门】组合+判断素数

题目描述

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

输出

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

样例输入

4 3
3 7 12 19

样例输出

1

#include <bits/stdc++.h>
using namespace std;
int p[30]={0},vis[30]={0};
int n,r,in[20]={0},cnt=0;
int judge(int n)
{
    if(n<=1)
        return 0;
    int k=sqrt(n);
    for(int i=2;i<k;i++)
    {
        if(n%i==0)
            return 0;
    }
    return 1;
}
void dfs(int cur,int &cnt)
{
  if(cur==r)
  {
      int sum=0;
      for(int i=0;i<r;i++)
      {

          sum+=p[i];
      }
      if(judge(sum))
      {
          cnt++;
      }

      return;
  }
  for(int i=1;i<=n;i++)
  {

      if(vis[i]==0&&(cur==0||p[cur-1]<in[i]))
      {
          p[cur]=in[i];
          vis[i]=1;
          dfs(cur+1,cnt);
          vis[i]=0;
      }
  }

}
int main()
{
     cin>>n>>r;
     for(int i=1;i<=n;i++)
        cin>>in[i];
     dfs(0,cnt);
     cout<<cnt;
    return 0;
}

答案错误33%，实在找不出来问题出在哪里

以下为copy的代码：

//i数组下标 第一个数后至少r-k个数 
//第k个位置放a[i]  j：开始选的第一个数 a数组中第j个数作为k位置第一个选项 
//搜索有范围  保证j后有足够的数 
#include<cstdio>
#include<cmath>//#include<numeric>
using namespace std;
int n,r,a[25],b[25],s,ans;
 
bool isss(int x){
	if (x<2) return false;
	int k = sqrt(x);
	for (int i=2;i<=k;i++)//2开始 
		if (x%i==0)
			return false;
	return true;
}
void dfs(int k,int j){
	if(k==r+1)
	{   s=0;//初始化!!!!  
		for(int i=1;i<=r;i++) s+=b[i];//t=accumulate(b+1,b+r+1,0);	
		if(isss(s)) ans++;
	    return ;
	}
	for(int i=j;i+r-k<=n;i++)
	{
		b[k]=a[i];
		dfs(k+1,i+1);//第k个数是a[i] 第k+1个数至少a[i+1] 
	}
}
int main(){
	scanf("%d%d",&n,&r);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	dfs(1,1);
	printf("%d",ans);
	return 0;
}