ツリー構造の子孫を全て取得する、エレガントな方法

Question

やりたいこと

開発中の Rails アプリで、このようなツリー構造のレコードがあり、その子孫を全て一次元の配列に格納したいと思っています。

全くエレガントではない私のコード

@items = []
# .subs は、その子オブジェクトを全て取得します。

@item.subs.each do |i2|
  @items.push(i2)
  i2.subs.each do |i3|
    @items.push(i3)
    i3.subs.each do |i4|
      @items.push(i4)
      i4.subs.each do |i5|
        @items.push(i5)
      end
    end
  end
end

@items.map(&:id) # [73, 170, 173, 175, 177, 178, 176, 174, 171, 172]

エレガントではないポイント
0. 階層が増えると、コードが増える。
0. ほぼ同じ処理を何度も書いている。
0. 再帰的な処理なので、while とかを使えそうなのに、上手く使えない無力さが如実に表れている。

お聞きしたいこと

同じ処理を、エレガントに記述する方法をご教示いただけないでしょうか。
何卒、よろしくお願いいたします。

Answer 1

配下の全レコードを集めるほうが、よりシンプルで、使い勝手も良さそう。

# インスタンスメソッドとして実現
class Record
  def all_records
    [self, *subs.flat_map(&:all_records)]
  end
end

@item.all_records.map(&:id)

# 関数として実現
def collect_records(recode)
  [
    recode,
    *recode.subs.flat_map { |sub_recode| collect_records(sub_recode) }
  ]
end

collect_records(@item).map(&:id)

Comments

[taichi730]

rubyっぽいので、こっちの方が好み

[todayske]

見るからにエレガントなコードで、ぜひ試してみたいのですが、理解できませんでした。（すみません・・・）
こちらの、:all_nodes はどんな引数なのでしょうか？

[taichi730]

:all_records のタイポです。
(修正しました。)
下位のレコードが持っているレコードを集めて(all_records/flat_map で実現)、１つの配列に纏めています。

[taichi730]

関数として実装した場合も追加しました。
やっていることは、両方とも同じです。

[todayske]

ご修正ありがとうございます！ (自分で気付くべきでした、大変失礼しました)
驚くほど美しくなって感動しております。
エレガントなコードに、重ねまして感謝です。ありがとうございます。

Answer 2

def scan_tree(node, &block)
  block.call(node)
  node.subs.each do |sub_node|
    scan_tree(sub_node, &block)
  end
end

@id = []
scan_tree(@item) do |node|
  id << node.id if node.respond_to?(:id)
end

こんな感じですかね？
元データ (@item) がないので、動作確認はしていませんが。

Comments

[todayske]

エ、エレガントだ・・・。
ほぼいただいたコードで動作確認できました！

-------------------------
def scan_tree(node, &block)
block.call(node)
node.subs.each do |sub_node|
scan_tree(sub_node, &block)
end
end

@items = []
scan_tree(@item) do |node|
@items << node
end
-------------------------

[todayske]

ありがとうございます！