結果

問題 No.1000 Point Add and Array Add
ユーザー simansiman
提出日時 2020-02-28 22:13:40
言語 Ruby
(3.3.0)
結果
WA  
実行時間 -
コード長 987 bytes
コンパイル時間 389 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 34,816 KB
最終ジャッジ日時 2024-04-21 19:06:14
合計ジャッジ時間 12,399 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 93 ms
12,032 KB
testcase_01 AC 89 ms
12,160 KB
testcase_02 WA -
testcase_03 AC 88 ms
12,288 KB
testcase_04 WA -
testcase_05 AC 87 ms
12,032 KB
testcase_06 AC 88 ms
12,032 KB
testcase_07 AC 88 ms
12,288 KB
testcase_08 AC 90 ms
12,288 KB
testcase_09 AC 89 ms
12,160 KB
testcase_10 AC 90 ms
12,032 KB
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 AC 757 ms
34,816 KB
testcase_21 AC 1,064 ms
34,816 KB
testcase_22 WA -
testcase_23 WA -
権限があれば一括ダウンロードができます
コンパイルメッセージ
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, -1)
      bit.add(x + 1, 1)
      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