Блог

Учитывая два входных аргумента:

x : lt;listgt; как [0, 1, 2, 3]

splits : lt;listgt; int представления числа и длины разбиений, например [1, 2, 1] . Это означает, что список должен быть разделен на 3, 1-й содержит 1 элемент, 2-й содержит 2 элемента, а 3-й раздел содержит 1 элемент

Мне нужна функция, которая возвращает все возможные комбинации расщеплений, например:

def get_combos(x, splits): ...

gt;gt;gt; get_combos([0, 1, 2, 3], [1, 2, 1])

 [(0,), (1, 2), (3,)] [(0,), (1, 3), (2,)] [(0,), (2, 3), (1,)] [(1,), (0, 2), (3,)] [(1,), (0, 3), (2,)] [(1,), (2, 3), (0,)] [(2,), (0, 1), (3,)] [(2,), (0, 3), (1,)] [(2,), (1, 3), (0,)] [(3,), (0, 1), (2,)] [(3,), (0, 2), (1,)] [(3,), (1, 2), (0,)]  

Входные данные всегда будут соответствовать этим условиям:

• min(splits) gt;= 1 —gt; В разделении всегда есть элемент
• sum(splits) == len(x) —gt; Все элементы в gt; x взяты

Каков наилучший способ сделать это рекурсивно?

Вы можете использовать рекурсивную функцию генератора:

 def combos(a, b, c = []):  if not a or not b:  yield [*map(tuple, c)]  else:  for x, y in enumerate(a):  if not c or len(c[-1]) == b[0]:  yield from combos(a[:x] a[x 1:], b[1:] if c else b, c [[y]])  else:  yield from combos(a[:x] a[x 1:], b, [*c[:-1], c[-1] [y]])  print(list(combos([0, 1, 2, 3], [1, 2, 1])))  

Выход:

 [[(0,), (1, 2), (3,)], [(0,), (1, 3), (2,)], [(0,), (2, 1), (3,)], [(0,), (2, 3), (1,)], [(0,), (3, 1), (2,)], [(0,), (3, 2), (1,)], [(1,), (0, 2), (3,)], [(1,), (0, 3), (2,)], [(1,), (2, 0), (3,)], [(1,), (2, 3), (0,)], [(1,), (3, 0), (2,)], [(1,), (3, 2), (0,)], [(2,), (0, 1), (3,)], [(2,), (0, 3), (1,)], [(2,), (1, 0), (3,)], [(2,), (1, 3), (0,)], [(2,), (3, 0), (1,)], [(2,), (3, 1), (0,)], [(3,), (0, 1), (2,)], [(3,), (0, 2), (1,)], [(3,), (1, 0), (2,)], [(3,), (1, 2), (0,)], [(3,), (2, 0), (1,)], [(3,), (2, 1), (0,)]]