aboutsummaryrefslogtreecommitdiff
path: root/day09/part2.rb
blob: 30fe1dbe8a4fc2ad91c091ef2e69d68438ceafd1 (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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
class Pos
  def initialize(x = 0, y = 0)
    @x = x
    @y = y
  end

  attr_reader :x, :y

  def move(dir)
    case dir
    when "U"
      @y += 1
    when "D"
      @y -= 1
    when "R"
      @x += 1
    when "L"
      @x -= 1
    end
  end

  def follow(head)
    if    head.x == @x - 2 && head.y == @y + 2
      @x -= 1
      @y += 1
    elsif head.x == @x - 1 && head.y == @y + 2
      @x -= 1
      @y += 1
    elsif head.x == @x     && head.y == @y + 2
      @y += 1
    elsif head.x == @x + 1 && head.y == @y + 2
      @x += 1
      @y += 1

    elsif head.x == @x + 2 && head.y == @y + 2
      @x += 1
      @y += 1
    elsif head.x == @x + 2 && head.y == @y + 1
      @x += 1
      @y += 1
    elsif head.x == @x + 2 && head.y == @y
      @x += 1
    elsif head.x == @x + 2 && head.y == @y - 1
      @x += 1
      @y -= 1

    elsif head.x == @x + 2 && head.y == @y - 2
      @x += 1
      @y -= 1
    elsif head.x == @x + 1 && head.y == @y - 2
      @x += 1
      @y -= 1
    elsif head.x == @x     && head.y == @y - 2
      @y -= 1
    elsif head.x == @x - 1 && head.y == @y - 2
      @x -= 1
      @y -= 1

    elsif head.x == @x - 2 && head.y == @y - 2
      @x -= 1
      @y -= 1
    elsif head.x == @x - 2 && head.y == @y - 1
      @x -= 1
      @y -= 1
    elsif head.x == @x - 2 && head.y == @y
      @x -= 1
    elsif head.x == @x - 2 && head.y == @y + 1
      @x -= 1
      @y += 1
    end
  end

  def coord
    [@x, @y]
  end
end

moves = $stdin.readlines.map(&:strip).map(&:split)

rope = Array.new(10) { Pos.new }
visited = [rope[-1].coord]

moves.each do |dir, dist|
  dist.to_i.times do
    rope[0].move(dir)
    (rope.length - 1).times do |i|
      rope[i+1].follow(rope[i])
    end
    visited << rope[-1].coord
  end
end

puts visited.uniq.count