This commit is contained in:
Meatball
2022-12-05 21:28:16 +01:00
committed by BethanyG
parent 2c102dcef1
commit dcfe63766c
2 changed files with 4 additions and 15 deletions

View File

@@ -2,12 +2,12 @@ import itertools
def combinations(target, size, exclude): def combinations(target, size, exclude):
result = [] result = []
possible = [i for i in range(1, target) if i not in exclude]
if size == 1: if size == 1:
return [[target]] return [[target]]
else: else:
for i in range(len(possible), 0, -1): possible = [index for index in range(1, int((target ** 2 / size) ** 0.6)) if index not in exclude]
for seq in itertools.combinations(possible, i): for index in range(len(possible), 0, -1):
for seq in itertools.combinations(possible, index):
if sum(seq) == target and len(seq) == size: if sum(seq) == target and len(seq) == size:
result.append(list(seq)) result.append(list(seq))
return result return result

View File

@@ -1,13 +1,2 @@
import itertools
def combinations(target, size, exclude): def combinations(target, size, exclude):
result = [] pass
if size == 1:
return [[target]]
else:
possible = [index for index in range(1, int((target ** 2 / size) ** 0.6)) if index not in exclude]
for index in range(len(possible), 0, -1):
for seq in itertools.combinations(possible, index):
if sum(seq) == target and len(seq) == size:
result.append(list(seq))
return result