結果
問題 |
No.3078 Difference Sum Query
|
ユーザー |
|
提出日時 | 2025-03-28 21:59:29 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 870 ms / 2,000 ms |
コード長 | 2,003 bytes |
コンパイル時間 | 261 ms |
コンパイル使用メモリ | 82,740 KB |
実行使用メモリ | 187,364 KB |
最終ジャッジ日時 | 2025-03-28 21:59:47 |
合計ジャッジ時間 | 16,509 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 26 |
ソースコード
import sys input = lambda :sys.stdin.readline()[:-1] ni = lambda :int(input()) na = lambda :list(map(int,input().split())) yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES") no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO") ####################################################################### # Binary Indexed Tree (Fenwick Tree) # 0-indexed class BIT: def __init__(self, n): self.n = n self.data = [0]*(n+1) def sum(self, i): s = 0 while i > 0: s += self.data[i] i -= i & -i return s def add(self, i, x): assert 0 <= i < self.n i = i+1 while i <= self.n: self.data[i] += x i += i & -i def set(self, i, x): assert 0 <= i <= self.n self.add(i,x-self.get(i)) def get(self, i, j): return self.sum(j) - self.sum(i) def lower_bound(self, k):# a[0],…,a[i]>kなる最小のi n = self.n x = 0 r = 1 while r <= n:r <<= 1 size = r while size: if x + size <= n and self.data[x+size] <= k: k-=self.data[x+size] x += size size >>= 1 return x def debug(self): res = [] n = self.n for i in range(n): res.append(self.get(i)) print(*res) n, q = na() a = na() que = [[] for i in range(n)] s = list(a) for i in range(q): l, r, x = na() l -= 1 r -= 1 if l: que[l-1].append((i, 1, x)) que[r].append((i, -1, x)) s.append(x) s = sorted(set(s)) m = len(s) d = {x:i for i,x in enumerate(s)} bs = BIT(m) bc = BIT(m) ans = [0] * q A = 0 N = 0 for i in range(n): bs.add(d[a[i]], a[i]) bc.add(d[a[i]], 1) A += a[i] N += 1 for j, t, x in que[i]: c = bc.get(d[x], m) s = bs.get(d[x], m) ans[j] -= t * (2 * s - A + (N - 2 * c) * x) for i in range(q): print(ans[i])