幸运的袋子（深度优先遍历（Depth First Search，DFS））

题目描述

一个袋子里面有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

 

分析

　　这道题涉及深度优先遍历。

　　对于任意两个正整数a,b如果满足 a+b>a*b，则必有一个数为1.可用数论证明：  设a=1+x,b=1+y，则1+x+1+y>(1+x)*(1+y)，--->  1>x*y，则x，y必有一个为0，即a,b有一个为1.  推广到任意k个正整数，假设a1,a2,...ak，如果不满足给定条件，即和sum小于等于积pi，  如果此时再选择一个数b,能使其满足sum+b > pi*b，则，b必然为1，且为必要非充分条件。  反之，如果选择的b>1，则sum+b <=pi*b，即a1,a2,...,ak,b不满足给定条件。（搜索剪枝的重要依据）　　因此，将球按标号升序排序 ，每次从小到大选择，当选择到a1,a2,...,ak-1时满足给定条件，而再增加选择ak时不满足条件（ak必然大于等于max(a1,a2,...,ak-1)），继续向后选择更大的数，必然无法满足。因此，可以进行剪枝。如果有多个1，即当k=1时，sum(1)>pi(1)不满足，但下一个元素仍为1，则可以满足1+1>1*1，所以要判断当前ak是否等于1。此外，对于重复数字，要去重复。  

 

上图即为剪枝后的部分解答树，每个结点即代表一个可能解，只有橙色节点为我们需要的解，例如最左边节点代表1 + 1 + 2 + 3 > 1 * 1 * 2

 

def getNum(array,Sum,multi):
    count = 0
    for i in range(len(array)):
        if i > 0 and array[i] == array[i-1]:
            continue
        Sum += array[i]
        multi *= array[i]
        if Sum > multi:
            count += 1+getNum(array[i+1:],Sum,multi)
        elif array[i] == 1:
            count += getNum(array[i+1:],Sum,multi)
        else:
            break
        Sum -= array[i]
        multi /= array[i]
    return count

n = input()
x = list(map(int,input().strip().split()))
x.sort()
Sum = 0
multi = 1
print(getNum(x,Sum,multi))