結果

問題 No.789 範囲の合計
ユーザー simansiman
提出日時 2022-11-25 02:59:09
言語 Ruby
(3.3.0)
結果
AC  
実行時間 525 ms / 1,000 ms
コード長 1,111 bytes
コンパイル時間 242 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 28,288 KB
最終ジャッジ日時 2024-10-01 14:38:26
合計ジャッジ時間 7,355 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 73 ms
12,032 KB
testcase_01 AC 74 ms
12,160 KB
testcase_02 AC 498 ms
24,448 KB
testcase_03 AC 410 ms
21,120 KB
testcase_04 AC 467 ms
25,216 KB
testcase_05 AC 481 ms
24,320 KB
testcase_06 AC 483 ms
24,576 KB
testcase_07 AC 399 ms
22,400 KB
testcase_08 AC 453 ms
28,288 KB
testcase_09 AC 452 ms
25,728 KB
testcase_10 AC 525 ms
22,784 KB
testcase_11 AC 490 ms
24,448 KB
testcase_12 AC 447 ms
25,216 KB
testcase_13 AC 75 ms
12,032 KB
testcase_14 AC 74 ms
12,160 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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
Q = N.times.map { gets.split.map(&:to_i) }

memo = Hash.new
values = []
Q.select { |t, _, _| t == 0 }.each do |t, x, y|
  values << x
end

values << Float::INFINITY
values.uniq!
values.sort!
values.each.with_index(1) do |v, idx|
  memo[v] = idx
end

bit = BinaryIndexTree.new(N + 2)
ans = 0

N.times do |i|
  if Q[i][0] == 0
    _, x, y = Q[i]
    idx = memo[x]
    bit.add(idx, y)
  else
    _, lv, rv = Q[i]
    l = values.bsearch_index { |v| v >= lv }
    r = values.bsearch_index { |v| v > rv }
    l += 1

    ans += bit.sum(0, r + 1) - bit.sum(0, l)
  end
end

puts ans
0