結果

問題 No.3078 Difference Sum Query
ユーザー tassei903
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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])
0