Tworzenie tablicy liczb, które sumują się pod podanym numerem

Question

Pracuję nad szybkimi i brudnymi skryptami do wykonywania niektórych zadań domowych z chemii, a jeden z nich przechodzi przez listy o stałej długości, w których sumują się wszystkie elementy dana stała. Dla każdego sprawdzam, czy spełniają niektóre dodatkowe kryteria i umieszczają je na innej liście.

ja wymyśliliśmy sposób, aby spełnić kryteria Sum, ale wygląda przerażające i jestem pewien, że istnieje jakiś rodzaj pojętny chwili tutaj:

# iterate through all 11-element lists where the elements sum to 8. 
for a in range(8+1): 
for b in range(8-a+1): 
    for c in range(8-a-b+1): 
    for d in range(8-a-b-c+1): 
    for e in range(8-a-b-c-d+1): 
    for f in range(8-a-b-c-d-e+1): 
     for g in range(8-a-b-c-d-e-f+1): 
     for h in range(8-a-b-c-d-e-f-g+1): 
     for i in range(8-a-b-c-d-e-f-g-h+1): 
     for j in range(8-a-b-c-d-e-f-g-h-i+1): 
      k = 8-(a+b+c+d+e+f+g+h+i+j) 
      x = [a,b,c,d,e,f,g,h,i,j,k] 
      # see if x works for what I want 

Answer 1

Oto generator rekurencyjny, który dostarcza list w porządku leksykograficznym. Pozostawienie exact jako True daje żądany wynik, gdzie każda suma == limit; ustawienie exact na False podaje wszystkie listy z 0 < = sumą < = limit. Rekurencja korzysta z tej opcji, aby uzyskać wyniki pośrednie.

def lists_with_sum(length, limit, exact=True): 
    if length: 
     for l in lists_with_sum(length-1, limit, False): 
      gap = limit-sum(l) 
      for i in range(gap if exact else 0, gap+1): 
       yield l + [i] 
    else: 
     yield [] 

Answer 2

Generic, rekurencyjne rozwiązanie:

def get_lists_with_sum(length, my_sum): 
    if my_sum == 0: 
     return [[0 for _ in range(length)]] 

    if not length: 
     return [[]] 
    elif length == 1: 
     return [[my_sum]] 
    else: 
     lists = [] 
     for i in range(my_sum+1): 
      rest = my_sum - i 
      sublists = get_lists_with_sum(length-1, rest) 
      for sl in sublists: 
       sl.insert(0, i) 
       lists.append(sl) 

    return lists 

print get_lists_with_sum(11, 8)