[2016][ruby][1] refactoring
diff --git a/2016/ruby/day_01.rb b/2016/ruby/day_01.rb
index cf606b2..552f04b 100644
--- a/2016/ruby/day_01.rb
+++ b/2016/ruby/day_01.rb
@@ -1,39 +1,49 @@
include Math
-def follow_instructions(input)
- seq = input.split(', ')
+Instruction = Struct.new(:direction, :blocks) do
+ def initialize(direction, blocks)
+ super(direction, blocks.to_i)
+ end
+end
- # start facing north at the origin
- dir = -PI/2
- locations = [[0,0]]
- seq.each.with_object(locations) do |step, locations|
- /(?<turn>\w)(?<blocks>\d+)/ =~ step
+Turtle = Struct.new(:orientation, :location) do
+ def self.run(input)
+ instructions = parse(input)
+ turtles = [Turtle.new(-PI/2, [0,0])]
- case turn
- when ?R
- dir += PI/2
- when ?L
- dir -= PI/2
- else
- raise "unexpected turn: #{turn}"
+ instructions.each.with_object(turtles) do |instruction, turtles|
+ turtles << turtles.last.turn(instruction.direction).step
+ (instruction.blocks - 1).times do
+ turtles << turtles.last.step
+ end
end
- blocks.to_i.times do
- location = locations.last
- x = location[0] + cos(dir).to_i
- y = location[1] + sin(dir).to_i
- locations << [x, y]
- end
+ turtles
end
- locations
-end
+ def self.parse(input)
+ input.split(', ').map {|step|
+ Instruction.new(*step.scan(/(?<turn>\w)(?<blocks>\d+)/)[0])
+ }
+ end
+
+ def turn(direction)
+ angle = case direction
+ when ?R
+ PI/2
+ when ?L
+ -PI/2
+ else
+ raise "unexpected direction: #{direction}"
+ end
+ Turtle.new(orientation + angle, location)
+ end
-def first_dupe(ary)
- seen = Set.new
- ary.find { |elem|
- seen.include?(elem) || (seen << elem && false)
- }
+ def step
+ x = location[0] + cos(orientation).to_i
+ y = location[1] + sin(orientation).to_i
+ Turtle.new(orientation, [x, y])
+ end
end
require 'minitest'
@@ -45,23 +55,17 @@ class TestInstructions < Minitest::Test
assert_distance 12, 'R5, L5, R5, R3'
end
- def test_first_dupe
- assert_equal 1, first_dupe([1, 2, 3, 1])
- assert_equal 2, first_dupe([1, 2, 3, 2])
- assert_equal 3, first_dupe([1, 2, 3, 3])
-
- locations = follow_instructions('R8, R4, R4, R8')
- assert_equal [4, 0], first_dupe(locations)
- end
-
def assert_distance(expected, instructions)
- locations = follow_instructions(instructions)
- assert_equal expected, locations.last.map(&:abs).inject(:+)
+ assert_equal expected, Turtle.run(instructions).last.location.map(&:abs).inject(:+)
end
end
if __FILE__ == $0
- puts first_dupe(follow_instructions(DATA.read))
+ turtles = Turtle.run(DATA.read)
+ p turtles.last
+
+ locations = turtles.map(&:location)
+ p locations.find {|location| locations.count(location) > 1 }
end
__END__