Функция перевода для нахождения всех разделов набора из Python в Ruby

Question

У меня есть следующая функция python для рекурсивного поиска всех разделов набора:

def partitions(set_):
    if not set_:
        yield []
        return
    for i in xrange(2**len(set_)/2):
        parts = [set(), set()]
        for item in set_:
            parts[i&1].add(item)
            i >>= 1
        for b in partitions(parts[1]):
            yield [parts[0]]+b

for p in partitions(["a", "b", "c", "d"]):
print(p)

Может кто-нибудь помочь мне перевести это на рубин? Это то, что я до сих пор:

def partitions(set)
  if not set
    yield []
    return
  end
  (0...2**set.size/2).each { |i|
    parts = [Set.new, Set.new]
    set.each { |item|
      parts[i&1] << item
      i >>= 1
    }
    partitions(parts[1]).each { |b|
      yield [parts[0]] << b
    }
  }
end

p partitions([1, 2, 3, 4].to_set)

Я получаю ошибку «LocalJumpError: блок не указан». Я думаю, это потому, что функции yield работают по-разному в Python и Ruby.

Answer 1

#!/usr/bin/ruby1.8

def partitions(set)
  yield [] if set.empty?
  (0 ... 2 ** set.size / 2).each do |i|
    parts = [[], []]
    set.each do |item|
      parts[i & 1] << item
      i >>= 1
    end
    partitions(parts[1]) do |b|
      result = [parts[0]] + b
      result = result.reject do |e|
        e.empty?
      end
      yield result
    end
  end
end

partitions([1, 2, 3, 4]) do |e|
  p e
end

# => [[1, 2, 3, 4]]
# => [[2, 3, 4], [1]]
# => [[1, 3, 4], [2]]
# => [[3, 4], [1, 2]]
# => [[3, 4], [2], [1]]
# => [[1, 2, 4], [3]]
# => [[2, 4], [1, 3]]
# => [[2, 4], [3], [1]]
# => [[1, 4], [2, 3]]
# => [[1, 4], [3], [2]]
# => [[4], [1, 2, 3]]
# => [[4], [2, 3], [1]]
# => [[4], [1, 3], [2]]
# => [[4], [3], [1, 2]]
# => [[4], [3], [2], [1]]

Что отличается:

• Охранник вызывает set.empty? вместо (неявно) тестирование для set.nil?
• Не указывать .each при звонке Перегородки
• Использовать массив вместо Set
• Фильтр пустых наборов из полученного Результат

Answer 2

Вам придется думать о yield Руби как о вызове пользовательской операции.

def twice
    yield
    yield
end

twice { puts "Hello" } 

Таким образом, всякий раз, когда ваш код возвращает значение, вызывается функция обработки для этого элемента.

partitions([1, 2, 3, 4].to_set) { |result| 
    # process result
}

Этот код вообще не генерирует список.