結果

問題 No.610 区間賞(Section Award)
ユーザー siman
提出日時 2023-06-09 01:39:31
言語 Ruby
(3.4.1)
結果
AC  
実行時間 521 ms / 2,000 ms
コード長 814 bytes
コンパイル時間 229 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 36,608 KB
最終ジャッジ日時 2024-12-31 09:15:28
合計ジャッジ時間 15,073 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 50
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

class BinaryIndexTree
  def initialize(size, init: 0)
    @values = Array.new(size, init)
    @size = size
  end

  # @param idx [Integer]
  # @param x [Numeric]
  def add(idx, x)
    raise 'Out of range reference' if @size <= idx

    idx += 1

    while idx <= @size
      @values[idx - 1] += x
      idx += idx & -idx
    end
  end

  def sum(l, r)
    _sum(r) - _sum(l)
  end

  private

  def _sum(idx)
    res = 0

    while idx > 0
      res += @values[idx - 1]
      idx -= idx & -idx
    end

    res
  end
end

N = gets.to_i
A = gets.split.map(&:to_i)
B = gets.split.map(&:to_i)

rank = Hash.new
A.each.with_index(1) do |a, i|
  rank[a] = i
end

bit = BinaryIndexTree.new(N + 1)
ans = []

B.each do |b|
  br = rank[b]

  if bit.sum(br, N + 1) == 0
    ans << b
  end

  bit.add(br, 1)
end

puts ans.sort
0