結果
問題 | No.877 Range ReLU Query |
ユーザー |
![]() |
提出日時 | 2019-09-06 22:23:08 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 606 ms / 2,000 ms |
コード長 | 1,819 bytes |
コンパイル時間 | 223 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 106,172 KB |
最終ジャッジ日時 | 2024-11-08 10:06:34 |
合計ジャッジ時間 | 7,792 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 20 |
ソースコード
class Bit: def __init__(self, n): self.size = n self.tree = [0]*(n+1) def __iter__(self): psum = 0 for i in range(self.size): csum = self.sum(i+1) yield csum - psum psum = csum raise StopIteration() def __str__(self): # O(nlogn) return str(list(self)) def sum(self, i): # [0, i) の要素の総和を返す #if not (0 <= i <= self.size): raise ValueError("error!") s = 0 while i>0: s += self.tree[i] i -= i & -i return s def add(self, i, x): #if not (0 <= i < self.size): raise ValueError("error!") i += 1 while i <= self.size: self.tree[i] += x i += i & -i def __getitem__(self, key): if not (0 <= key < self.size): raise IndexError("error!") return self.sum(key+1) - self.sum(key) def __setitem__(self, key, value): # 足し算と引き算にはaddを使うべき if not (0 <= key < self.size): raise IndexError("error!") self.add(key, value - self[key]) from operator import itemgetter N, Q = map(int, input().split()) A = list(map(int, input().split())) bit = Bit(N) bit2 = Bit(N) for i, a in enumerate(A): bit.add(i, a) Ans = [-1] * Q ILRX = [] A_ = sorted(enumerate(A), key=itemgetter(1)) for i in range(Q): q, l, r, x = map(int, input().split()) ILRX.append((i, l-1, r, x)) ILRX.sort(key=itemgetter(3)) idx_a_ = 0 for i, l, r, x in ILRX: while idx_a_ < N: idx_a, a = A_[idx_a_] if a < x: bit.add(idx_a, -a) bit2.add(idx_a, 1) idx_a_ += 1 else: break Ans[i] = bit.sum(r)-bit.sum(l) - (r-l-bit2.sum(r)+bit2.sum(l))*x print("\n".join(map(str, Ans)))