結果
| 問題 |
No.1000 Point Add and Array Add
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2021-08-21 19:08:34 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 334 ms / 2,000 ms |
| コード長 | 894 bytes |
| コンパイル時間 | 159 ms |
| コンパイル使用メモリ | 82,208 KB |
| 実行使用メモリ | 104,576 KB |
| 最終ジャッジ日時 | 2024-10-15 09:44:38 |
| 合計ジャッジ時間 | 5,586 ms |
|
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 22 |
ソースコード
import sys
input = sys.stdin.readline
N, Q = map(int, input().split())
A = list(map(int, input().split()))
class Bit:
"""1-indexed"""
def __init__(self, n):
self.size = n
self.tree = [0] * (n + 1)
def sum(self, i):
s = 0
while i > 0:
s += self.tree[i]
i -= i & -i
return s
def add(self, i, x):
while i <= self.size:
self.tree[i] += x
i += i & -i
bit = Bit(N + 1)
ans = [0] * N
for _ in range(Q):
c, x, y = input().split()
x, y = int(x), int(y)
if c == 'A':
n = bit.sum(x)
ans[x - 1] += n * A[x - 1]
bit.add(x, -n)
bit.add(x + 1, n)
A[x - 1] += y
else:
# assert c == 'B'
bit.add(x, 1)
bit.add(y + 1, -1)
for x in range(1, N + 1):
n = bit.sum(x)
ans[x - 1] += n * A[x - 1]
print(*ans)