回溯求解子集

对于给定一个集合,求解它的子集,或者根据条件筛选子集,可以利用回溯来很好的解决这个问题。

例如(洛谷P1036)

题目如下:

已知 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
解题思路:利用一个数组来记录元素状态,当元素状态都标记时,就可以进行判断筛选。

#include<iostream>
#include<cmath>
using namespace std;
int group[21],a[21],b[5000001];
int count,sum,ans=0,length=0;
void back(int x1,int x2,int k){
	if(x1>x2){
		count=0;
		sum=0;
		for(int i=1;i<=x2;i++){
			if(a[i]){
				count++;
				sum+=group[i];
			}
		}
		if(count==k)
			b[length++]=sum;
		return ;
	}
	else{
		a[x1]=1;
		back(x1+1,x2,k);
		a[x1]=0;
		back(x1+1,x2,k);
	}
	return ;
}
int prime(int x){
	int i;
	if(x<2)
		return 0;
	for(i=2;i*i<=x;i++){
		if(x%i==0)
			return 0;
	}
	return 1;
}
int main(){
	int n,k;
	cin>>n>>k;
	for(int i=1;i<=n;i++)
		cin>>group[i];
	back(1,n,k);
	for(int i=0;i<length;i++)
		if(prime(b[i]))
			ans++;
	cout<<ans;
	return 0;
}



  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值