結果
問題 |
No.610 区間賞(Section Award)
|
ユーザー |
![]() |
提出日時 | 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
ソースコード
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