[2019][ruby][11.x] use trig

master
Alpha Chen 5 years ago
parent 10ace7d99a
commit 975834ce55

@ -2,16 +2,17 @@ require "stringio"
require_relative "computer" require_relative "computer"
include Math
panels = Hash.new {|h,k| h[k] = [0]} panels = Hash.new {|h,k| h[k] = [0]}
panels[[0, 0]] = [1] panels[[0, 0]] = [1]
i_r, i_w = IO.pipe i_r, i_w = IO.pipe
o_r, o_w = IO.pipe o_r, o_w = IO.pipe
Robot = Struct.new(:panel, :dir) Robot = Struct.new(:panel, :rad)
dirs = %i[ up right down left ] robot = Robot.new([0, 0], PI/2)
robot = Robot.new([0, 0], 0)
t = Thread.new { t = Thread.new {
loop do loop do
@ -19,22 +20,11 @@ t = Thread.new {
panels[robot.panel] << o_r.gets.to_i panels[robot.panel] << o_r.gets.to_i
dir = o_r.gets.to_i dir = o_r.gets.to_i
raise "unexpected direction: #{dir}" unless [0, 1].include?(dir)
robot.rad += -(2*dir - 1) * (PI/2)
delta = case dir delta = [cos(robot.rad).to_i, sin(robot.rad).to_i]
when 0 then -1
when 1 then 1
else raise "unexpected direction: #{dir}"
end
robot.dir += delta
robot.dir %= dirs.size
dir = dirs.fetch(robot.dir)
delta = case dir
when :up then [ 0, 1]
when :right then [ 1, 0]
when :down then [ 0, -1]
when :left then [-1, 0]
end
robot.panel = robot.panel.zip(delta).map(&:sum) robot.panel = robot.panel.zip(delta).map(&:sum)
end end
} }

Loading…
Cancel
Save