結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 89 ms
12,416 KB
testcase_01 AC 91 ms
12,288 KB
testcase_02 AC 90 ms
12,160 KB
testcase_03 AC 91 ms
12,160 KB
testcase_04 AC 91 ms
12,160 KB
testcase_05 AC 89 ms
12,160 KB
testcase_06 AC 90 ms
12,160 KB
testcase_07 AC 90 ms
12,032 KB
testcase_08 AC 90 ms
12,160 KB
testcase_09 AC 89 ms
12,288 KB
testcase_10 AC 89 ms
12,288 KB
testcase_11 AC 89 ms
12,160 KB
testcase_12 AC 100 ms
12,416 KB
testcase_13 AC 100 ms
12,544 KB
testcase_14 AC 102 ms
12,416 KB
testcase_15 AC 101 ms
12,416 KB
testcase_16 AC 853 ms
32,640 KB
testcase_17 AC 813 ms
24,576 KB
testcase_18 AC 1,217 ms
34,688 KB
testcase_19 AC 1,226 ms
34,688 KB
testcase_20 AC 758 ms
34,816 KB
testcase_21 AC 1,035 ms
34,816 KB
testcase_22 AC 1,264 ms
34,688 KB
testcase_23 AC 1,094 ms
34,816 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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