blob: b021d10e36b6d2d087e9fdd6f8258c12acb67903 (
plain)
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#!/usr/bin/env ruby
class Machine
def initialize(starting_memory = [])
@mem = starting_memory
@pc = 0
@halt = false
@buffered_input = []
@buffered_output = []
end
def mem(start = 0, len = @mem.length)
@mem[start...(start + len)]
end
def halt!
@halt = true
end
def halt?
@halt
end
def buffer_input(x)
@buffered_input.push(x)
end
def buffered_output
t = @buffered_output
@buffered_output = []
t
end
def step
return if halt?
opcode = @mem[@pc] % 100
p3, p2, p1 = ("%03d" % (@mem[@pc] / 100)).chars.map{|c|c == "1"}
l1w = @mem[@pc + 1]
l1r = p1 ? l1w : @mem[l1w]
l2w = @mem[@pc + 2]
l2r = p2 ? l2w : @mem[l2w]
l3w = @mem[@pc + 3]
l3r = p3 ? l3w : @mem[l3w]
case opcode
when 1
@mem[l3w] = l1r + l2r
@pc += 4
when 2
@mem[l3w] = l1r * l2r
@pc += 4
when 3
@mem[l1w] = @buffered_input.shift
@pc += 2
when 4
@buffered_output.push(l1r)
@pc += 2
when 99
halt!
@pc += 1
end
end
def run
puts buffered_output
until halt? do
step
puts buffered_output unless @buffered_output.empty?
end
end
end
input = $stdin.readlines[0].strip.split(",").map(&:to_i)
m = Machine.new(input)
m.buffer_input(1)
m.run
|
