結果

問題 No.789 範囲の合計
ユーザー simansiman
提出日時 2022-11-25 02:59:09
言語 Ruby
(3.1.1p18 )
結果
AC  
実行時間 541 ms / 1,000 ms
コード長 1,111 Byte
コンパイル時間 78 ms
使用メモリ 29,400 KB
最終ジャッジ日時 2022-11-25 02:59:19
合計ジャッジ時間 7,800 ms
ジャッジサーバーID
(参考情報)
judge13 / judge11
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 62 ms
13,964 KB
testcase_01 AC 61 ms
13,832 KB
testcase_02 AC 480 ms
25,940 KB
testcase_03 AC 397 ms
22,212 KB
testcase_04 AC 486 ms
26,892 KB
testcase_05 AC 472 ms
26,032 KB
testcase_06 AC 480 ms
26,124 KB
testcase_07 AC 405 ms
23,024 KB
testcase_08 AC 446 ms
29,400 KB
testcase_09 AC 432 ms
26,988 KB
testcase_10 AC 541 ms
25,108 KB
testcase_11 AC 454 ms
26,744 KB
testcase_12 AC 429 ms
26,748 KB
testcase_13 AC 66 ms
13,724 KB
testcase_14 AC 70 ms
13,844 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