|
|
@ -59,12 +59,37 @@ class Spiral
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Grid
|
|
|
|
|
|
|
|
def initialize
|
|
|
|
|
|
|
|
@grid = Hash.new {|h,(x,y)|
|
|
|
|
|
|
|
|
neighbors = [[-1, 1], [0, 1], [1, 1],
|
|
|
|
|
|
|
|
[-1, 0], [1, 0],
|
|
|
|
|
|
|
|
[-1,-1], [0,-1], [1,-1]]
|
|
|
|
|
|
|
|
h[[x,y]] = neighbors
|
|
|
|
|
|
|
|
.map {|i,j| [x+i,y+j] }
|
|
|
|
|
|
|
|
.select {|i,j| h.has_key?([i,j]) }
|
|
|
|
|
|
|
|
.map {|i,j| h[[i,j]] }
|
|
|
|
|
|
|
|
.sum
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
@grid[[0,0]] = 1
|
|
|
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def [](x,y)
|
|
|
|
|
|
|
|
@grid[[x,y]]
|
|
|
|
|
|
|
|
end
|
|
|
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
if $0 == __FILE__
|
|
|
|
if $0 == __FILE__
|
|
|
|
if ARGV.shift == "test"
|
|
|
|
case ARGV.shift
|
|
|
|
require "minitest/test"
|
|
|
|
when "test"
|
|
|
|
else
|
|
|
|
require "minitest/autorun"
|
|
|
|
|
|
|
|
when "0"
|
|
|
|
input = ARGF.read.to_i
|
|
|
|
input = ARGF.read.to_i
|
|
|
|
p Spiral.new.each.with_index.find {|_,i| i == input - 1 }.first.sum
|
|
|
|
p Spiral.new.each.with_index.find {|_,i| i == input - 1 }.first.sum
|
|
|
|
|
|
|
|
when "1"
|
|
|
|
|
|
|
|
input = ARGF.read.to_i
|
|
|
|
|
|
|
|
grid = Grid.new
|
|
|
|
|
|
|
|
p Spiral.new.each.lazy.map {|x,y| grid[x,y] }.find {|value| value > input }
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
@ -87,3 +112,15 @@ class TestSpiral < Minitest::Test
|
|
|
|
assert_equal [0,-2], spiral.each.with_index.find {|_,i| i == 22 }.first
|
|
|
|
assert_equal [0,-2], spiral.each.with_index.find {|_,i| i == 22 }.first
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestGrid < Minitest::Test
|
|
|
|
|
|
|
|
def test_grid
|
|
|
|
|
|
|
|
grid = Grid.new
|
|
|
|
|
|
|
|
assert_equal 1, grid[0,0]
|
|
|
|
|
|
|
|
assert_equal 1, grid[1,0]
|
|
|
|
|
|
|
|
assert_equal 2, grid[1,1]
|
|
|
|
|
|
|
|
assert_equal 4, grid[0,1]
|
|
|
|
|
|
|
|
assert_equal 5, grid[-1,1]
|
|
|
|
|
|
|
|
assert_equal 10, grid[-1,0]
|
|
|
|
|
|
|
|
end
|
|
|
|
|
|
|
|
end
|
|
|
|