输入两个整数n和m,从数列1,2,3,。。。n中随意取几个数,使其和等于m,请把所有可能的组合都列出来;
例如:
n=5,m=6
那么:
1 5
2 4
1 2 3
答案:(没有参照任何人,自己想出来的)
import time
n = 20
m = 20
a = time.time()
def get_list(x, y, result, result_list=[]):
'递归得到 x+1 到 y 之间 和 等于 y 的所有组合 '
for j in range(x + 1, y // 2 + 1):
y2 = y - j
if y2 <= j:
y = y2
continue
if j in result_list or y2 in result_list:
continue
result_list.append(j)
result_list.append(y2)
if sum(result2) == m:
result3 = sorted(result_list[:]) # 排序,之后好进行判断
if result3 not in result:
result.append(result3)
result_list.pop() # 删除最后一位,用这里的最后一位继续循环
get_list(x, y2, result, result_list)
result_list.pop() # 再删除最后一位,用剩下的数继续循环
print(a)
result = []
for i in range(1, m // 2 + 1): # range(1,6)
y = m - i
if y > n:
continue
if y > i:
result.append([i, y])
x = i
result2 = [i]
get_list(x, y, result, result2)
print(result)
print(len(result))
print(time.time())
print(time.time() - a)