結果

問題 No.2065 Sum of Min
ユーザー ニックネーム
提出日時 2022-09-02 22:47:32
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 945 bytes
コンパイル時間 149 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 166,856 KB
最終ジャッジ日時 2024-11-16 05:14:02
合計ジャッジ時間 51,451 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 3 TLE * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

class BIT:
    def __init__(self, n):
        self.n = n; self.k = [0]*(n+1)
    def a(self, i, x):
        while i <= self.n: self.k[i] += x; i += i&-i
    def s(self, i):
        t = 0
        while i > 0: t += self.k[i]; i -= i&-i
        return t
    def r(self, l, r): return self.s(r)-self.s(l-1)
n, q = map(int, input().split())
a = list(map(int, input().split()))
lrmpxi = []
vxset = set(a)
for i in range(q):
    l, r, x = map(int, input().split())
    lrmpxi.append((l-2, -1, x, i))
    lrmpxi.append((r-1, 1, x, i))
    vxset.add(x)
vxtoi = {vx: i+1 for i, vx in enumerate(sorted(vxset))}
bit_sum = BIT(n+q+1)
bit_num = BIT(n+q+1)
j = -1
ans = [0]*q
for lr, mp, x, i in sorted(lrmpxi):
    while j < lr:
        j += 1
        v = a[j]
        bit_sum.a(vxtoi[v], v)
        bit_sum.a(n+q+1, -v)
        bit_num.a(1, 1)
        bit_num.a(vxtoi[v], -1)
    ans[i] += mp*(bit_sum.s(vxtoi[x])+x*bit_num.s(vxtoi[x]))
print(*ans, sep="\n")
0