编程题——幸运袋子
题目描述:
一个袋子里面有n个球,每个球上面都有一个号码(拥有相同号码的球是无区别的)。如果一个袋子是幸运的当且仅当所有球的号码的和大于所有球的号码的积。
例如:如果袋子里面的球的号码是{1, 1, 2, 3},这个袋子就是幸运的,因为1 + 1 + 2 + 3 > 1 * 1 * 2 * 3
你可以适当从袋子里移除一些球(可以移除0个,但是别移除完),要使移除后的袋子是幸运的。现在让你编程计算一下你可以获得的多少种不同的幸运的袋子。
输入描述:
第一行输入一个正整数n(n ≤ 1000)
第二行为n个数正整数xi(xi ≤ 1000)
输出描述:
输出可以产生的幸运的袋子数
示例1:
输入
3
1 1 1
输出
2
示例2:
输入
6
1 1 2 4 1 2
输出
12
程序代码如下:
#include <iostream>
#include <algorithm>
using namespace std;
/*这道题实际上是求符合条件的子集的个数
*要保证正整数号码和大于积,则必须有一个数是1
*所以我们首先将数组升序排列,然后循环读取号码进行判断:
* 以1作为入口向后组合进行判断
* 如果和大于积,向后加1位继续判断,
* 直到不满足条件时,跳出循环,再回退到当前位(此时count已+1),替换当前位的号码,重新判断新组合
* 对于相同的号码,直接跳过避免重复组合*/
int GetLuckyBag(int x[], int n, int pos, int sum, int mul)
{
int count = 0;
for (int i = pos; i < n; ++i)
{
sum += x[i];
mul *= x[i];
if (sum>mul)
count += 1 + GetLuckyBag(x, n, i + 1, sum, mul);
else if (1 == x[i])
count += GetLuckyBag(x, n, i + 1, sum, mul);
else
break;
sum -= x[i];
mul /= x[i];
while (i < n - 1 && x[i] == x[i + 1])
++i;
}
return count;
}
int main()
{
int n;
int x[1000] = { 0 };
while (cin >> n)
{
x[1000] = { 0 };
for (int i = 0; i < n; ++i)
cin >> x[i];
sort(x, x + n);
cout << GetLuckyBag(x, n, 0, 0, 1)<<endl;
}
return 0;
}
程序运行结果如下: