子集和问题(回溯法)

题目

设集合S={x1,x2,…,xn}是一个正整数集合，c是一个正整数，子集和问题判定是否存在S的一个子集S1，使S1中的元素之和为c。试设计一个解子集和问题的回溯法，并输出利用回溯法在搜索树（按输入顺序建立）中找到的第一个解。

输入格式:
输入数据第1行有2个正整数n和c，n表示S的大小，c是子集和的目标值。接下来的1行中，有n个正整数，表示集合S中的元素。 是子集和的目标值。接下来的1 行中，有n个正整数，表示集合S中的元素。

输出格式:
输出利用回溯法找到的第一个解，以空格分隔，最后一个输出的后面有空格。当问题无解时，输出“No Solution!”。

输入样例:
在这里给出一组输入。例如：

5 10
2 2 6 5 4

输出样例:
在这里给出相应的输出。例如：

2 2 6 

答案

#include<iostream>
using namespace std;
int n,c,tsum=0,total=0;
int num[1001],vis[1001];
bool sum(int i)
{
	if(tsum==c) return true;
	if(i>n) return false;
	total-=num[i];
	if(tsum+num[i]<=c)
	{
		vis[i]=1;
		tsum+=num[i];
		if(sum(i+1)) return true;
		tsum-=num[i];
	}
	if(tsum+total>=c)
	{
		vis[i]=0;
		if(sum(i+1)) return true;
	}
	total+=num[i];
	return false;
}
int main()
{
	cin>>n>>c;
	for(int i=1;i<=n;i++)
	{
		cin>>num[i];
		total+=num[i];
	}
	if(!sum(1)) cout<<"No Solution!";
	else
	{
		for(int i=1;i<=n;i++)
		if(vis[i]) cout<<num[i]<<" ";
	}
}

参考

理解回溯法及例题分析