最近看的一道算法题,算是常规题目了,用python实现了一下
思路:用二进制的0000000000~1111111111代表某个数字是否被选中,一个位代表数组中的一个元素是否被选中,可以概括所有的情况,如果数字是0101010101代表45,99,6,20,18五个数字被选出来了。接着我们只需要计算着五个数是否等于我们要最终需要sum
data_list = [-10,45,35,99,10,6,9,20,17,18]
def calSum(data_list, result):
for i in range(1<<len(data_list)):
sum = 0
tmp = str()
for j in range(len(data_list)):
if i & 1 << j:
sum = sum + data_list[j]
tmp = str(data_list[j]) if tmp == '' else tmp + ' + ' + str(data_list[j])
if sum == result:
print(i, bin(i))
print(tmp + ' = 35')
calSum(data_list, 35)
结果输出如下
$ python3.6 sum_m.py
3 0b11
-10 + 45 = 35
4 0b100
35 = 35
21 0b10101
-10 + 35 + 10 = 35
224 0b11100000
6 + 9 + 20 = 35
241 0b11110001
-10 + 10 + 6 + 9 + 20 = 35
768 0b1100000000
17 + 18 = 35
785 0b1100010001
-10 + 10 + 17 + 18 = 35