結果
問題 |
No.1000 Point Add and Array Add
|
ユーザー |
![]() |
提出日時 | 2020-02-28 21:46:06 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 470 ms / 2,000 ms |
コード長 | 1,493 bytes |
コンパイル時間 | 184 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 139,652 KB |
最終ジャッジ日時 | 2024-10-13 17:12:42 |
合計ジャッジ時間 | 6,138 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
class BIT(): """区間加算、一点取得クエリをそれぞれO(logN)で答える add: 区間[l, r)にvalを加える get_val: i番目の値を求める i, l, rは0-indexed """ def __init__(self, n): self.n = n self.bit = [0] * (n + 1) def _add(self, i, val): while i > 0: self.bit[i] += val i -= i & -i def get_val(self, i): """i番目の値を求める""" i = i + 1 s = 0 while i <= self.n: s += self.bit[i] i += i & -i return s def add(self, l, r, val): """区間[l, r)にvalを加える""" self._add(r, val) self._add(l, -val) n, q = list(map(int, input().split())) a = list(map(int, input().split())) query = [list(input().split()) for i in range(q)] bit = BIT(n) imos = [0] * (n + 1) for i in range(q): if query[i][0] == "B": _, l, r = query[i] l = int(l) - 1 r = int(r) - 1 bit.add(l, r + 1, 1) imos[l] += 1 imos[r + 1] -= 1 for i in range(n): imos[i + 1] += imos[i] ans = [0] * n for i in range(n): ans[i] += a[i] * imos[i] for i in range(q): if query[i][0] == "A": _, ind, val = query[i] ind = int(ind) - 1 val = int(val) ans[ind] += val * bit.get_val(ind) if query[i][0] == "B": _, l, r = query[i] l = int(l) - 1 r = int(r) - 1 bit.add(l, r + 1, -1) print(*ans)