結果

問題 No.877 Range ReLU Query
ユーザー H3PO4H3PO4
提出日時 2021-03-04 11:55:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 586 ms / 2,000 ms
コード長 1,043 bytes
コンパイル時間 303 ms
コンパイル使用メモリ 82,372 KB
実行使用メモリ 108,544 KB
最終ジャッジ日時 2024-04-25 22:30:34
合計ジャッジ時間 7,343 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 33 ms
53,540 KB
testcase_01 AC 45 ms
63,784 KB
testcase_02 AC 45 ms
64,404 KB
testcase_03 AC 51 ms
67,892 KB
testcase_04 AC 39 ms
59,560 KB
testcase_05 AC 43 ms
63,236 KB
testcase_06 AC 43 ms
62,608 KB
testcase_07 AC 41 ms
61,096 KB
testcase_08 AC 51 ms
68,528 KB
testcase_09 AC 42 ms
59,984 KB
testcase_10 AC 46 ms
63,708 KB
testcase_11 AC 520 ms
102,816 KB
testcase_12 AC 473 ms
99,232 KB
testcase_13 AC 383 ms
95,472 KB
testcase_14 AC 419 ms
96,712 KB
testcase_15 AC 586 ms
108,544 KB
testcase_16 AC 583 ms
103,264 KB
testcase_17 AC 565 ms
104,804 KB
testcase_18 AC 569 ms
106,844 KB
testcase_19 AC 528 ms
101,848 KB
testcase_20 AC 577 ms
106,208 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

input = sys.stdin.buffer.readline


class Bit:
    """1-indexed"""

    def __init__(self, n):
        self.size = n
        self.tree = [0] * (n + 1)

    def _sum(self, i):
        s = 0
        while i > 0:
            s += self.tree[i]
            i -= i & -i
        return s

    def sum(self, i, j):
        """閉区間[i, j]"""
        return self._sum(j) - self._sum(i - 1)

    def add(self, i, x):
        while i <= self.size:
            self.tree[i] += x
            i += i & -i


N, Q = map(int, input().split())
A = tuple(map(int, input().split()))
SA = [(a, i) for i, a in enumerate(A, 1)]
SA.sort()
queries = []
for i in range(Q):
    _, l, r, x = map(int, input().split())
    queries.append((i, l, r, x))
queries.sort(key=lambda t: -t[3])

bit_alive = Bit(N)
bit_sum = Bit(N)
ans = [0] * Q
for i, l, r, x in queries:
    while SA and SA[-1][0] >= x:
        a, ia = SA.pop()
        bit_alive.add(ia, 1)
        bit_sum.add(ia, a)
    ans[i] = bit_sum.sum(l, r) - bit_alive.sum(l, r) * x
print(*ans, sep='\n')
0