我刚开始Backtracking
在大学学习算法。不知何故,我设法为子集和问题制作了一个程序。工作正常,但后来我发现我的程序并没有给出所有可能的组合。
例如:目标总和可能有一百个组合,但我的程序只给出 30 个。这是代码。如果有人能指出我的错误是什么,那将是一个很大的帮助。
int tot=0;//tot is the total sum of all the numbers in the set.
int prob[500], d, s[100], top = -1, n; // n = number of elements in the set. prob[i] is the array with the set.
void subset()
{
int i=0,sum=0; //sum - being updated at every iteration and check if it matches 'd'
while(i<n)
{
if((sum+prob[i] <= d)&&(prob[i] <= d))
{
s[++top] = i;
sum+=prob[i];
}
if(sum == d) // d is the target sum
{
show(); // this function just displays the integer array 's'
top = -1; // top points to the recent number added to the int array 's'
i = s[top+1];
sum = 0;
}
i++;
while(i == n && top!=-1)
{
sum-=prob[s[top]];
i = s[top--]+1;
}
}
}
int main()
{
cout<<"Enter number of elements : ";cin>>n;
cout<<"Enter required sum : ";cin>>d;
cout<<"Enter SET :\n";
for(int i=0;i<n;i++)
{
cin>>prob[i];
tot+=prob[i];
}
if(d <= tot)
{
subset();
}
return 0;
}
当我运行程序时:
Enter number of elements : 7
Enter the required sum : 12
Enter SET :
4 3 2 6 8 12 21
SOLUTION 1 : 4, 2, 6
SOLUTION 2 : 12
虽然 4、8 也是一种解决方案,但我的程序并没有显示出来。输入数量为 100 或更多时,情况更糟。至少会有 10000 个组合,但我的程序显示 100 个。
我试图遵循的逻辑:
- 只要子集的总和保持小于或等于目标总和,就将主 SET 的元素纳入子集。
- 如果将特定数字添加到子集总和使其大于目标,则不接受它。
- 一旦它到达集合的末尾,并且没有找到答案,它就会从集合中删除最近使用的数字,并开始查看最近删除的数字位置之后的位置中的数字。(因为我存储在数组's'中的是主SET中所选数字的位置)。