采用回溯法。 首先对输入的球的号码num[]进行从小到大排序。设置result[]数组用于记录被选取的球的号码。用1表示选取某个球,0表示不选取,解是0和1的组合,组成了一个高度为n+1的解空间树,左子树表示1,右子树表示0。在解空间树中,第t层结点表示是否选取第t+1个球(t从0开始),如果选取,则往左子树扩展,并且把该球的号码加入num[]数组中。如果不选取,则往右子树扩展。当目前的结点和所要判断的号码和其双亲结点相同,并且是双亲结点的右孩子时,则只往右子树扩展。当到达叶子结点时,则判断num[]数组里面的和是否大于积,如果大于,则幸运袋子数加一。
#include <iostream>
using namespace std;
int lucky = 0; //幸运袋子数
struct result_list
{
int result[100];
int num_of_result = 0;
}result; //记录选择的结果
void sort(int num[], int n) //排序
{
for (int i = 0; i < n - 1; i++)
{
int min = i, tmp;
for (int j = i + 1; j < n; j++)
{
if (num[j] < num[min])
min = j;
}
tmp = num[i];
num[i] = num[min];
num[min] = tmp;
}
}
void append(int n_append) //在result.result[]末尾增加一个元素n_append
{
result.result[result.num_of_result] = n_append;
result.num_of_result++;
}
void pop() //去掉result.result[]末尾的一个元素
{
result.result[result.num_of_result - 1] = 0;
result.num_of_result--;
}
void traceback(int t, int last, int flag, int num[], int n)
{
int sum = 0, mul = 1;
if (t == n) //遍历到叶子结点时
{
int i = 0;
while (result.result[i] != 0) //计算和与积
{
sum += result.result[i];
mul *= result.result[i];
i++;
}
if (sum > mul)
lucky += 1;
sum = 0; //得到一个幸运袋子后把sum和mul重新设为初始值
mul = 1;
}
else if (t < n) //还没到叶子结点时
{
if (num[t] == last && flag == 0) //如果此时正在判断的数和上一个数相同,并且目前是双亲结点的右孩子时
{
traceback(t + 1, num[t], 0, num, n); //只深度优先遍历右子树
}
else
{
append(num[t]);
traceback(t + 1, num[t], 1, num, n); //深度优先遍历左子树
pop();
traceback(t + 1, num[t], 0, num, n); //深度优先遍历右子树
}
}
}
int main(void)
{
int n = 0;
cin >> n; //输入n
int* num = new int[n];
for (int i = 0; i < n; i++) //输入num[]
{
cin >> num[i];
}
cout << endl;
sort(num, n); //先对num[]进行从小到大排序
traceback(0, 0, 1, num, n);
cout << lucky << endl;
return 0;
}
运行结果: