ruby初心者です。階層の深いhashの作り方

Question

いま、深い階層を作成しようとしています。
しかし、途中で行き詰まってしまいどうしていいかわからなくなってしましました。

h_symptom=Hash.new
symptom=Array.new
cause1=Array.new
cause2=Array.new

a_line.each{|dset|
	if dset[0]==3 then
		h_symptom[dset[1]]=Hash.new
		symptom=dset[1]
	elsif dset[0]==6 then
		h_symptom[symptom][dset[1]]=Hash.new
		cause1=dset[1]
	elsif dset[0]==9 then
		h_symptom[symptom][cause1][dset[1]]=Hash.new
		cause2=dset[1]
	elsif dset[0]==12 then
		h_symptom[symptom][cause1][cause2] = dset[1]
	end
}

dset[0]には３の倍数が入っており、さらに深い階層にも対応させたいです。

3,a
6,b
9,c
12,d
15,e
18,f
3、あ
6,い

a_lineはこんな感じになっています。
３から次の３までを階層にしたい場合、どうしたら簡略的に書くことが出来ますでしょうか？

よろしくお願いします。

Comments

[mpyw]

本題とは無関係ですが

symptom=Array.new
cause1=Array.new
cause2=Array.new

これいらないです．ここで作ったインスタンスが結局

symptom=dset[1]

こういった処理に全て上書きされています．

Answer 1

次の機能を実装してみました。

• ネストレベルが０番でないものが複数並ぶ場合を処理できるようにする。
• 入力データの数字が 3 9 と飛んでいたらエラーにする。
• ネストしたハッシュを入力データの形式で表示する。

aa.rb

require 'awesome_print'  # gem install awesome_print

# 質問文にあったデータ例
data0 = <<'EOS'
3,a
6,b
9,c
12,d
15,e
18,f
3,あ
6,い
EOS

# 追加したデータ例
data1 = <<'EOS'
3,a
6,b
9,c
3,あ
6,い
9,1
9,2
6,う
9,10
9,20
EOS

# 不正なデータ例
dataX = <<'EOS'
3,a
9,c
EOS

# [nest, key] の組に配列データからネストしたハッシュを生成する
def make_hash(lines)
  ans = {}
  keys = []
  lines.each do |line|
    pos = line[0] / 3 - 1
    fail "Bad data: #{line}" if pos > keys.length

    keys = (pos == 0) ? [] : keys[0..(pos - 1)]
    keys << line[1]
    keys.inject(h = ans) do |h, key|
      h[key] = {} unless h[key]
      h[key]
    end
  end
  ans
end

def print_keys(hash, nest = 0)
  hash.each do |key, val|
    puts "#{(nest + 1) * 3} #{key}"
    print_keys(val, nest + 1) unless val.empty?
  end
end

a_lines = data0.lines.map do |line|
  x = line.chop.split(',')
  [x[0].to_i, x[1..-1].join(' ')]
end
hash = make_hash(a_lines)
ap hash
p hash
print_keys(hash)
puts

a_lines = data1.lines.map do |line|
  x = line.chop.split(',')
  [x[0].to_i, x[1..-1].join(' ')]
end
hash = make_hash(a_lines)
ap hash
print_keys(hash)
puts

a_lines = dataX.lines.map do |line|
  x = line.chop.split(',')
  [x[0].to_i, x[1..-1].join(' ')]
end
hash = make_hash(a_lines)
# ap hash

実行結果

$ ruby aa.rb
{
    "a" => {
        "b" => {
            "c" => {
                "d" => {
                    "e" => {
                        "f" => {}
                    }
                }
            }
        }
    },
    "あ" => {
        "い" => {}
    }
}
{"a"=>{"b"=>{"c"=>{"d"=>{"e"=>{"f"=>{}}}}}}, "あ"=>{"い"=>{}}}
3 a
6 b
9 c
12 d
15 e
18 f
3 あ
6 い

{
    "a" => {
        "b" => {
            "c" => {}
        }
    },
    "あ" => {
        "い" => {
            "1" => {},
            "2" => {}
        },
        "う" => {
            "10" => {},
            "20" => {}
        }
    }
}
3 a
6 b
9 c
3 あ
6 い
9 1
9 2
6 う
9 10
9 20

aa.rb:41:in `block in make_hash': Bad data: [9, "c"] (RuntimeError)
	from aa.rb:39:in `each'
	from aa.rb:39:in `make_hash'
	from aa.rb:83:in `<main>'

Answer 2

こういう書き方は，**「どこまでがキーでどこからが値なのか」**が不明瞭になるので(競技プログラミングでこういう問題が用意されている場合を除き)設計ミスです．葉要素の値は空ハッシュにしてしまうほうがいいと思います．

entries = [
  [3, :a],
  [6, :b],
  [9, :c],
  [12, :d],
  [15, :e],
  [18, :f],
  [3, :あ],
  [6, :い],
]

root = {}
entries.inject(root) do |node, entry|
  node = root if entry[0] == 3
  node[entry[1]] = {} if node[entry[1]].nil?
  node[entry[1]]
end

p root
# {:a=>{:b=>{:c=>{:d=>{:e=>{:f=>{}}}}}}, :あ=>{:い=>{}}}