[2018][ruby][09.2]

sorbet
Alpha Chen 6 years ago
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…
Cancel
Save