【洛谷】过程函数与递归 P1036 选数

题目描述

已知 nnn 个整数 x1,x2,…,xnx_1,x_2,…,x_nx1​,x2​,…,xn​,以及111个整数kkk(k<nk<nk<n)。从nnn个整数中任选kkk个整数相加,可分别得到一系列的和。例如当n=4,k=3n=4,k=3n=4,k=3,444个整数分别为3,7,12,193,7,12,193,7,12,19时,可得全部的组合与它们的和为:

3+7+12=223+7+12=223+7+12=22

3+7+19=293+7+19=293+7+19=29

7+12+19=387+12+19=387+12+19=38

3+12+19=343+12+19=343+12+19=34。

现在,要求你计算出和为素数共有多少种。

例如上例,只有一种的和为素数:3+7+19=293+7+19=293+7+19=29。
输入格式

键盘输入,格式为:

n,kn,kn,k(1≤n≤20,k<n1 \le n \le 20,k<n1≤n≤20,k<n)

x1,x2,…,xn(1≤xi≤5000000)x_1,x_2,…,x_n (1 \le x_i \le 5000000)x1​,x2​,…,xn​(1≤xi​≤5000000)
输出格式

屏幕输出,格式为: 111个整数(满足条件的种数)。

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

输出 #1
1

#include <iostream>
#include <math.h> 
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
//想法: 利用深度搜索算法,每一个数都有选与不选两种选择DFS 、
using namespace std;
const int maxn=100;
int n , k ;   //总共n个数,选择其中三个,求和判断素数个数
int x[maxn]; 
int index=0,nowk=0,ans=0,sum=0;

//void Dfs(int index,int nowk,int  sum)
//{
//	if(index==n)
//	    {
//	    	if(nowk==k)
//	    	{
//	    		for(int i = 2 ; i <=sqrt(sum);i++)
//	    		{
//	    			if(sum%i == 0)
//	    			{
//	    				return ;
//					}
//				}
//				ans++;
//			}
//	    	return ;
//		}
//	
//	Dfs(index+1,nowk,sum);   //不选此数
//	Dfs(index+1,nowk+1,sum+x[index]);  //选此数 
//}

//对算法进行优化,1.如果nowk超过k直接return,不需要再进行递归!
// 2.利用筛法进行判断素数而不是开方方法 
void Dfs(int index,int nowk,int  sum)
{
	int status=0;
	if(index==n)
	{
		return ;
	 } 
	 
	 Dfs(index+1,nowk,sum);   //不选此数
	 
	if(nowk+1<=k)
	{
		if(nowk+1==k)
		{
	    for(int i = 2 ; i <=sqrt(sum+x[index]);i++)
	    {
	    	if((sum+x[index])%i == 0)
	    	{
	    		status=1;
	    		break;
			}	
		}
		
		if(status==0)
		{
		ans++;
		 
		} 
		
	}
	status=0;
		Dfs(index+1,nowk+1,sum+x[index]);  //选此数 
	}
	
	
}

int main(int argc, char** argv) {
	cin >> n >> k ;
	for(int i = 0 ; i < n ; i++)
	{
		cin >> x[i];
	 } 
	 //index,nowk,sum 
	Dfs(index,nowk,sum);
	cout << ans;
	return 0;
}

素数的判断还可以用筛法来降低复杂度

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
对于洛谷上的p1036题目,我们可以使用Python来解决。下面是一个可能的解法: ```python def dfs(nums, target, selected_nums, index, k, sum): if k == 0 and sum == target: return 1 if index >= len(nums) or k <= 0 or sum > target: return 0 count = 0 for i in range(index, len(nums)): count += dfs(nums, target, selected_nums + [nums[i]], i + 1, k - 1, sum + nums[i]) return count if __name__ == "__main__": n, k = map(int, input().split()) nums = list(map(int, input().split())) target = int(input()) print(dfs(nums, target, [], 0, k, 0)) ``` 在这个解法中,我们使用了深度优先搜索(DFS)来找到满足要求的数列。通过归的方式,我们遍历了所有可能的数字组合,并统计满足条件的个数。 首先,我们从给定的n和k分别表示数字个数和需要取的数字个数。然后,我们输入n个数字,并将它们存储在一个列表nums中。接下来,我们输入目标值target。 在dfs函数中,我们通过迭代index来择数字,并更新取的数字个数k和当前总和sum。如果k等于0且sum等于target,我们就找到了一个满足条件的组合,返回1。如果index超出了列表长度或者k小于等于0或者sum大于target,说明当前组合不满足要求,返回0。 在循环中,我们不断归调用dfs函数,将取的数字添加到selected_nums中,并将index和k更新为下一轮归所需的值。最终,我们返回所有满足条件的组合个数。 最后,我们在主程序中读入输入,并调用dfs函数,并输出结果。 这是一种可能的解法,但不一定是最优解。你可以根据题目要求和测试数据进行调试和优化。希望能对你有所帮助!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值