結果

問題 No.2065 Sum of Min
ユーザー tassei903tassei903
提出日時 2022-09-02 22:28:18
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,510 ms / 2,000 ms
コード長 1,964 bytes
コンパイル時間 153 ms
コンパイル使用メモリ 82,216 KB
実行使用メモリ 213,084 KB
最終ジャッジ日時 2024-04-27 22:04:40
合計ジャッジ時間 25,116 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
52,200 KB
testcase_01 AC 40 ms
52,752 KB
testcase_02 AC 40 ms
52,752 KB
testcase_03 AC 40 ms
53,440 KB
testcase_04 AC 670 ms
148,220 KB
testcase_05 AC 623 ms
148,316 KB
testcase_06 AC 1,131 ms
132,088 KB
testcase_07 AC 449 ms
130,388 KB
testcase_08 AC 1,106 ms
136,204 KB
testcase_09 AC 1,328 ms
212,872 KB
testcase_10 AC 1,356 ms
212,616 KB
testcase_11 AC 1,403 ms
212,840 KB
testcase_12 AC 1,444 ms
213,024 KB
testcase_13 AC 1,465 ms
212,088 KB
testcase_14 AC 1,437 ms
212,428 KB
testcase_15 AC 1,450 ms
212,608 KB
testcase_16 AC 1,446 ms
213,016 KB
testcase_17 AC 1,510 ms
212,684 KB
testcase_18 AC 1,456 ms
213,084 KB
testcase_19 AC 1,450 ms
212,484 KB
testcase_20 AC 1,442 ms
212,852 KB
testcase_21 AC 1,452 ms
213,048 KB
権限があれば一括ダウンロードができます

ソースコード

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)
        self.el = [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
        self.el[i] += x
        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=None):# j=Noneのときiを取得, [i, j)の和を取得
        if j is None:
            return self.el[i+1]
        return self.sum(j) - self.sum(i)


n, q = na()
a = na()
s = list(a)

event = []


for i in range(q):
    l,r,x = na()
    event.append((l-1,0,x, i))
    event.append((r-1,2,x, i))
    s.append(x)

s = sorted(set(s))

d = {x:i for i, x in enumerate(s)}
m = len(s)
bit1 = BIT(m)
bit2 = BIT(m)

for i in range(n):
    event.append((i, 1, a[i]))
ans = [0] * q
for eve in sorted(event):
    if eve[1] == 0:
        x = eve[2]
        i = eve[3]
        ans[i] -= bit2.get(0, d[x]) + x * bit1.get(d[x], m)
    elif eve[1] == 1:
        x = eve[2]
        bit1.add(d[x], 1)
        bit2.add(d[x], x)
    else:
        x = eve[2]
        i = eve[3]
        ans[i] += bit2.get(0, d[x]) + x * bit1.get(d[x], m)
    """print(eve)
    for j in range(m):
        print(bit1.get(j),end = " ")
    print()
    for j in range(m):
        print(bit2.get(j),end = " ")
    print()
    print(*ans)"""
for i in ans:
    print(i)
0