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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
require "z3"

Machine = Data.define(:lights, :schematics, :joltage) do
  def configure_lights
    schematics = self.schematics.map { it.inject(0) {|n,i| n + 2**i } }
    (1..).find {|n|
      schematics.repeated_combination(n).any? {|presses|
        presses.inject(0) {|lights, i| lights ^ i } == lights
      }
    }
  end

  def configure_joltage
    solver = Z3::Optimize.new

    schematics = self.schematics.map.with_index { [_1, Z3.Int("n#{_2}")] }

    solver.minimize(schematics.map(&:last).sum)

    schematics.each do |_, v|
      solver.assert(v >= 0)
    end

    joltage.each.with_index do |level, i|
      solver.assert(
        schematics.filter_map {|s,v| s.include?(i) && v }.inject(:+) == level
      )
    end

    fail unless solver.satisfiable?
    solver.model.to_h.values.sum(&:to_i)
  end
end

machines = DATA.read.lines.map {|line|
  lights, *schematics, joltage = line.split(/\s+/)

  lights = lights.scan(/[.#]/).map {
    case it
    when ?. then 0
    when ?# then 1
    else fail
    end
  }.join.reverse.to_i(2)
  schematics = schematics.map { it.scan(/\d+/).map(&:to_i) }
  joltage = joltage.scan(/\d+/).map(&:to_i)

  Machine.new(lights, schematics, joltage)
}

pp machines.map(&:configure_lights).sum
pp machines.map(&:configure_joltage).sum

__END__