it-swarm.dev

Python: использование рекурсивного алгоритма в качестве генератора

Недавно я написал функцию для генерации определенных последовательностей с нетривиальными ограничениями. Проблема пришла с естественным рекурсивным решением. Теперь случается так, что даже при относительно небольших входных данных последовательности составляют несколько тысяч, поэтому я предпочел бы использовать мой алгоритм в качестве генератора, а не использовать его для заполнения списка всеми последовательностями.

Вот пример. Предположим, мы хотим вычислить все перестановки строки с помощью рекурсивной функции. Следующий наивный алгоритм принимает дополнительный аргумент "storage" и добавляет к нему перестановку всякий раз, когда он его находит:

def getPermutations(string, storage, prefix=""):
   if len(string) == 1:
      storage.append(prefix + string)   # <-----
   else:
      for i in range(len(string)):
         getPermutations(string[:i]+string[i+1:], storage, prefix+string[i])

storage = []
getPermutations("abcd", storage)
for permutation in storage: print permutation

(Пожалуйста, не заботьтесь о неэффективности, это только пример.)

Теперь я хочу превратить свою функцию в генератор, то есть получить перестановку вместо добавления ее в список хранения:

def getPermutations(string, prefix=""):
   if len(string) == 1:
      yield prefix + string             # <-----
   else:
      for i in range(len(string)):
         getPermutations(string[:i]+string[i+1:], prefix+string[i])

for permutation in getPermutations("abcd"):
   print permutation

Этот код работает не (функция ведет себя как пустой генератор).

Я что-то пропустил? Есть ли способ превратить вышеуказанный рекурсивный алгоритм в генератор не заменяя его на итеративный?

99
Federico A. Ramponi
def getPermutations(string, prefix=""):
    if len(string) == 1:
        yield prefix + string
    else:
        for i in xrange(len(string)):
            for perm in getPermutations(string[:i] + string[i+1:], prefix+string[i]):
                yield perm

Или без аккумулятора:

def getPermutations(string):
    if len(string) == 1:
        yield string
    else:
        for i in xrange(len(string)):
            for perm in getPermutations(string[:i] + string[i+1:]):
                yield string[i] + perm
117
Markus Jarderot

Это позволяет избежать глубокой рекурсии len(string)- и, в общем, является хорошим способом обработки генераторов внутри генераторов:

from types import GeneratorType

def flatten(*stack):
    stack = list(stack)
    while stack:
        try: x = stack[0].next()
        except StopIteration:
            stack.pop(0)
            continue
        if isinstance(x, GeneratorType): stack.insert(0, x)
        else: yield x

def _getPermutations(string, prefix=""):
    if len(string) == 1: yield prefix + string
    else: yield (_getPermutations(string[:i]+string[i+1:], prefix+string[i])
            for i in range(len(string)))

def getPermutations(string): return flatten(_getPermutations(string))

for permutation in getPermutations("abcd"): print permutation

flatten позволяет нам продолжать прогресс в другом генераторе, просто yielding его, вместо итерации по нему и yielding каждого элемента вручную.


Python 3.3 добавит yield from к синтаксису, что позволяет естественное делегирование субгенератору:

def getPermutations(string, prefix=""):
    if len(string) == 1:
        yield prefix + string
    else:
        for i in range(len(string)):
            yield from getPermutations(string[:i]+string[i+1:], prefix+string[i])
28
ephemient

Внутренний вызов getPermutations - это тоже генератор.

def getPermutations(string, prefix=""):
   if len(string) == 1:
      yield prefix + string            
   else:
      for i in range(len(string)):
         getPermutations(string[:i]+string[i+1:], prefix+string[i])  # <-----

Вам нужно повторить это с помощью цикла for (см. Публикацию @MizardX, которая меня затянула на несколько секунд!)

19
S.Lott