'''數字組合問題'''n = 5r = 3a = [1,2,3,4,5] # 五個數字x = [0]*n # 一個解（n元0,1數組） 固定長度X = []  # 一組解def conflict(k):  global n, r, x  if sum(x[:k+1]) > r: # 部分解的長度超出r    return True  if sum(x[:k+1]) + (n-k-1) < r: # 部分解的長度加上剩下長度不夠r    return True  return False # 無沖突# 套用子集樹模板def comb(k): # 到達第k個元素  global n, x, X  if k >= n: # 超出最尾的元素    #print(x)    X.append(x[:]) # 保存（一個解）  else:    for i in [1, 0]: # 遍歷元素 a[k] 的兩種選擇狀態:1-選擇，0-不選      x[k] = i      if not conflict(k): # 剪枝        comb(k+1)# 根據一個解x，構造對應的一個組合def get_a_comb(x):  global a  return [y[0] for y in filter(lambda s:s[1]==1, zip(a, x))]# 根據一組解X，構造對應的一組組合def get_all_combs(X):  return [get_a_comb(x) for x in X]# 測試comb(0)print(X)print(get_all_combs(X))