设集合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<stdio.h>
#define Max 10000
int array[Max]={0};
int chose[Max]={0};
int sum=0;
int f=0;
void print(int x)
{
for(int i=0;i<=x;i++)
{
if(chose[i])
{
printf("%d ",array[i]);
}
}
}
int Backtrack(int x,int n,int c)
{
while(1)
{
if(!chose[x])
{
chose[x]=1;
sum+=array[x];
if(sum==c)
{
print(x);
return 1;
}
else if(sum>c)
{
chose[x]=0;
sum-=array[x];
}
x++;
}
if(x==n)
{
//因为每一个分支都是优先选择"加入的",因此每一层回溯的时候不用考虑“不加”的分支
while(1)
{
x--;
if(x<0)return 0;//无解
if(chose[x]==0)
break;
chose[x]=0;//将剪去的结点的分支默认都设置为0,以便后面重新开始
//一直回溯到选择“不加”的那一个结点,回溯路径上的“不加”的分支全部剪掉
}
//找到第一个“不加”分支的结点后,继续回溯,找到第一个“加”分支选择的结点
while(1)
{
x--;
if(x<0)return 0;//无解
if(chose[x])
break;
}
//x位置结点的“加”分支被全部剪掉,转而考虑不加“分支”
sum-=array[x];
chose[x]=0;
x++;
}
}
}
int main()
{
int n,c;
scanf("%d%d",&n,&c);
int total=0;
for(int i=0;i<n;i++)
{
scanf("%d",&array[i]);
total+=array[i];
}
f=Backtrack(0,n,c);
if(total<c||!f)
printf("No Solution!");
}