Выразительная простота python на примере задач из комбинаторики

В процессе самообучения языку программирования python (имея знания с/с++) решил написать в качестве задания функции генерирующие элементы из различных множеств комбинаторных конфигураций. Конечно, можно справедливо заметить, что подобный функционал уже есть в стандартной библиотеке python в модуле itertools, но у каждого должно быть право изобрести велосипед, тем более в целях обучения…Тот кто знаком с основами теории вероятностей должны помнить, что такое урновые схемы и о чем эта таблица: f2eb23a2f8484c1395b98ec1af9e370a.pngИ так ТЗ — написать четыре генератора, которые принимая строку s, состоящую из уникальных символов, и размер выборки к, возвращают строку — выборку с повторением/без повторений из k символов строки s порядок важен/не важен.В результате получился следующий код:

import math

def template (s, k, assertion = None, reducer = None): n = len (s) assert assertion (n, k) if k == 0: yield » return elif k == 1: for c in s: yield c return

k-=1 for i in range (n): c = s[i] new_s = reducer (s, i) if not assertion (len (new_s), k): break for res in template (new_s, k, assertion, reducer): yield c+res assertion_norep = lambda n, k: n > 0 and n >= k and k >= 0 assertion_rep = lambda n, k: n > 0 and k >= 0

def permutation_norep (s, k): return template (s, k, assertion = assertion_norep, reducer = lambda s, i: s[: i]+s[i+1:]) def permutation_rep (s, k): return template (s, k, assertion = assertion_rep, reducer = lambda s, i: s)

def combination_norep (s, k): return template (s, k, assertion = assertion_norep, reducer = lambda s, i: s[i+1:]) def combination_rep (s, k): return template (s, k, assertion = assertion_rep, reducer = lambda s, i: s[i:])

test = «abcdefg» k = 5 n = len (test) assert len (set (permutation_norep (test, k))) == (math.factorial (n) / math.factorial (n-k)) assert len (list (permutation_norep (test, k))) == (math.factorial (n) / math.factorial (n-k))

assert len (set (permutation_rep (test, k))) == n**k assert len (list (permutation_rep (test, k))) == n**k

assert len (set (combination_norep (test, k))) == (math.factorial (n) / (math.factorial (n-k)*math.factorial (k))) assert len (list (combination_norep (test, k))) == (math.factorial (n) / (math.factorial (n-k)*math.factorial (k)))

assert len (set (combination_rep (test, k))) == (math.factorial (n+k-1) / (math.factorial (n-1)*math.factorial (k))) assert len (list (combination_rep (test, k))) == (math.factorial (n+k-1) / (math.factorial (n-1)*math.factorial (k))) Так как python является языком еще более высокого уровня абстракции, чем с/с++, поэтому он позволяет проще и выразительнее писать код, который бы на других языках выглядел бы более громоздко и запутаннее. Новичкам в python я хотел бы обратить внимание на несколько моментов:

return после yield Рекурсивный генератор Шаблон стратегия Использование lambda функций P.S. Могу добавить, что я не сразу пришел к подобному решению, использующему общую «шаблонную» функцию. Сначала я написал все функции по отдельности, а потом выделил общее и различное.

© Habrahabr.ru