結果

問題 No.2942 Sigma Music Game Level Problem
ユーザー tassei903
提出日時 2024-10-18 21:55:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,854 ms / 6,000 ms
コード長 2,021 bytes
コンパイル時間 410 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 246,504 KB
最終ジャッジ日時 2024-11-15 15:58:13
合計ジャッジ時間 21,405 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

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

m = 2 * 10 ** 5 + 10

n, q, _ = na()
a = na()

bit1 = BIT(m)
bit2 = BIT(m)
FLAG = 0
for i in range(n):
    bit1.add(a[i], 1)
    bit2.add(a[i], a[i])

for _ in range(q):
    que = na()
    if que[0] == 1:
        l = que[1]
        bit1.add(l, 1)
        bit2.add(l, l)
    elif que[0] == 2:
        FLAG = 1
        l, r = que[1], que[2] + 1
        print(bit1.get(l, r), bit2.get(l, r))
    else:
        continue

if FLAG == 0:
    print("Not Found!")
0