AtCoder ABC127 D hash à résoudre avec Ruby 2.7.1

introduction

L'environnement des problèmes passés d'AtCoder a été mis à niveau.

Ce thème

AtCoder Beginner Contest D - Integer Cards Difficulty: 891

Ce thème, hash

D'après le problème, c'est une file d'attente prioritaire, mais il semble que les restrictions sont lâches et il peut être résolu par le tri.

Solution 1 Hash (2.7.1)

ruby.rb

n, m = gets.split.map(&:to_i)
h = gets.split.map(&:to_i).tally
m.times do
  b, c = gets.split.map(&:to_i)
  if h.key?(c)
    h[c] += b
  else
    h[c] = b
  end
end
ans = 0
cnt = 0
h.sort{_2 <=> _1}.each do |k, v|
  if cnt + v < n
    ans += k * v
    cnt += v
  else
    ans += k * (n - cnt)
    break
  end
end
puts ans

tally.rb

h = gets.split.map(&:to_i).tally

Le hachage est généré à partir du tableau avec «tally».

numberedparameter.rb

h.sort{_2 <=> _1}.each do |k, v|       # after 2.7.1

h.sort{|a, b| b <=> a}.each do |k, v|  # before

Vous pouvez spécifier les paramètres du bloc par numéro. La méthode de description de sort_by est inconnue.

Hachage de la solution 2

ruby.rb

n, m = gets.split.map(&:to_i)
a = gets.split.map(&:to_i)
h = Hash.new(0)
a.each do |x|
  h[x] += 1
end
m.times do
  b, c = gets.split.map(&:to_i)
  h[c] += b
end
ans = 0
cnt = 0
h.sort_by{|k, v| -k}.each do |k, v|
  if cnt + v < n
    ans += k * v
    cnt += v
  else
    ans += k * (n - cnt)
    break
  end
end
puts ans

C'est un style d'écriture qui peut être passé même avec 2.3.3.

Baie de la solution 3

ruby.rb

n, m = gets.split.map(&:to_i)
a = gets.split.map(&:to_i).sort
h = []
m.times do
  h << gets.split.map(&:to_i)
end
i = 0
h.sort_by{|u, v| -v}.each do |x, y|
  break if a[i] >= y
  x.times do
    a[i] = y
    i += 1
    break if i >= n
    break if a[i] >= y
  end
  break if i >= n
end
puts a.inject(:+)

Je l'ai fait passer vers le mois de mai. Il semble que ce soit un peu plus rapide si vous passez cette fois.

Solution 4 File d'attente prioritaire

ruby.rb

class Heap
  attr_reader :size
  def initialize(up: false)
    @up = up
    @heap = []
    @size = 0
  end
  def sum
    x = 0
    @size.times do |i|
      x += @heap[i]
    end
    x
  end
  def push(n)
    n = -n if @up
    i = @size
    while i > 0
      pid = (i - 1) / 2
      break if n >= @heap[pid]
      @heap[i] = @heap[pid]
      i = pid
    end
    @heap[i] = n
    @size += 1
  end
  def pop
    return nil if @size == 0
    top = @heap[0]
    @size -= 1
    n = @heap[@size]
    i = 0
    while i * 2 + 1 < @size
      cid1 = i * 2 + 1
      cid2 = cid1 + 1
      if cid2 < @size && @heap[cid2] < @heap[cid1]
        cid1 = cid2
      end
      break if @heap[cid1] >= n
      @heap[i] = @heap[cid1]
      i = cid1
    end
    @heap[i] = n
    if @up
      -top
    else
      top
    end
  end
end

_, m = gets.split.map(&:to_i)
a = gets.split.map(&:to_i)
b = Array.new(m){gets.split.map(&:to_i)}
h = Heap.new(up: false)
a.each do |x|
  h.push(x)
end
f = false
b.sort_by{|x, y| -y}.each do |x, y|
  x.times do
    u = h.pop
    if y > u
      h.push(y)
    else
      h.push(u)
      f = true
    end
    break if f
  end
  break if f
end
puts h.sum

Prend la valeur minimale du tas, la compare et la renvoie.

Le paramètre numéroté semble être utile.

Résumé

• Résolu ABC 127 D
• Familiarisez-vous avec Ruby

Site référencé

• [Comprenez avec un exemple de code! Principales nouvelles fonctionnalités et modifications de Ruby 2.7 Partie 1 - paramètre numéroté (https://qiita.com/jnchito/items/79f0172e60f237e2c542) *