day12.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

require 'json'

d = JSON.parse(File.read("day12"))

def sum(o)
  case o
  when Hash
    sum o.values
  when Array
    o.inject(0) { |a,e| a + sum(e) }
  when Numeric
    o
  else
    0
  end
end

def sum2(o)
  case o
  when Hash
    if o.values.include?("red")
      0
    else
      sum2 o.values
    end
  when Array
    o.inject(0) { |a,e| a + sum2(e) }
  when Numeric
    o
  else
    0
  end
end

p sum(d)

p sum2(d[2][7]["h"]["c"][0]) # 96852