parent
4576ad656c
commit
337fa5835b
@ -1,35 +1,45 @@
|
|||||||
|
Node = Struct.new(*%i[ value next prev ]) do
|
||||||
|
def to_s
|
||||||
|
out = [value]
|
||||||
|
current = self
|
||||||
|
loop do
|
||||||
|
current = current.next
|
||||||
|
break if current == self
|
||||||
|
out << current.value
|
||||||
|
end
|
||||||
|
out.join(" ")
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
ARGF.read.chomp =~ /(\d+) players; last marble is worth (\d+) points/
|
ARGF.read.chomp =~ /(\d+) players; last marble is worth (\d+) points/
|
||||||
num_players = $1.to_i
|
num_players = $1.to_i
|
||||||
num_marbles = $2.to_i
|
num_marbles = $2.to_i * 100
|
||||||
|
|
||||||
circle = []
|
current = Node.new(0, nil, nil)
|
||||||
marbles = (0...num_marbles).to_a
|
current.next = current.prev = current
|
||||||
|
|
||||||
zero = marbles.shift
|
|
||||||
circle << zero
|
|
||||||
|
|
||||||
current_marble_index = 0
|
|
||||||
current_player = 0
|
current_player = 0
|
||||||
scores = Hash.new(0)
|
scores = Array.new(num_players, 0)
|
||||||
|
|
||||||
until marbles.empty?
|
(1...num_marbles).each do |marble|
|
||||||
current_marble = marbles.shift
|
if marble % 23 == 0
|
||||||
|
scores[current_player] += marble
|
||||||
if current_marble % 23 == 0
|
7.times { current = current.prev }
|
||||||
scores[current_player] += current_marble
|
scores[current_player] += current.value
|
||||||
index = (current_marble_index - 7) % circle.size
|
current.next.prev = current.prev
|
||||||
scores[current_player] += circle.delete_at(index)
|
current.prev.next = current.next
|
||||||
current_marble_index = index
|
current = current.next
|
||||||
else
|
else
|
||||||
index = (current_marble_index + 2) % circle.size - 1
|
current = current.next
|
||||||
index += 1 unless index == -1
|
current.next = Node.new(marble, current.next, current)
|
||||||
circle.insert(index, current_marble)
|
current = current.next
|
||||||
current_marble_index = index % circle.size
|
current.next.prev = current
|
||||||
end
|
end
|
||||||
|
|
||||||
# puts "[#{current_player + 1}] #{circle.join(" ")}"
|
# puts "[#{current_player + 1}] #{current}"
|
||||||
|
puts marble if marble % 100000 == 0
|
||||||
|
|
||||||
current_player = (current_player + 1) % num_players
|
current_player = (current_player + 1) % num_players
|
||||||
end
|
end
|
||||||
|
|
||||||
p scores.max_by(&:last)
|
p scores.max
|
||||||
|
Loading…
Reference in new issue