結果

問題 No.1000 Point Add and Array Add
ユーザー siman
提出日時 2020-02-28 22:16:06
言語 Ruby
(3.4.1)
結果
AC  
実行時間 1,181 ms / 2,000 ms
コード長 991 bytes
コンパイル時間 105 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 34,688 KB
最終ジャッジ日時 2024-10-13 17:50:27
合計ジャッジ時間 11,633 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

class BinaryIndexTree
  attr_reader :bit, :size

  def initialize(size, init: 0)
    @bit = Array.new(size + 1, init)
    @size = size
  end

  def sum(i)
    s = 0

    while i > 0
      s += bit[i]
      i -= i & -i
    end

    s
  end

  def add(i, x)
    while i <= size
      bit[i] += x
      i += i & -i
    end
  end
end

N, Q = gets.split.map(&:to_i)
A = gets.split.map(&:to_i)
B = Array.new(N + 1, 0)
C = Array.new(N + 2, 0)
bit = BinaryIndexTree.new(N + 2)

Q.times do
  c, x, y = gets.chomp.split
  x = x.to_i
  y = y.to_i

  if c == 'A'
    cnt = bit.sum(x)

    # p [x, cnt]

    if cnt > 0
      # p 'middle update'
      B[x] += cnt * A[x - 1]
      # p B
      bit.add(x, -cnt)
      bit.add(x + 1, cnt)
      C[x] -= cnt
      C[x + 1] += cnt
    end

    A[x - 1] += y
  else
    bit.add(x, 1)
    bit.add(y + 1, -1)
    C[x] += 1
    C[y + 1] -= 1
  end
  
  # p [:C, C]
end

c = 0

1.upto(N) do |i|
  c += C[i]
  B[i] += A[i - 1] * c
end

# p A
puts B[1..-1].join(' ')
0