結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
52,224 KB
testcase_01 AC 55 ms
63,872 KB
testcase_02 AC 60 ms
62,976 KB
testcase_03 AC 68 ms
67,200 KB
testcase_04 AC 49 ms
59,008 KB
testcase_05 AC 56 ms
61,312 KB
testcase_06 AC 51 ms
61,696 KB
testcase_07 AC 50 ms
60,928 KB
testcase_08 AC 62 ms
67,200 KB
testcase_09 AC 45 ms
59,520 KB
testcase_10 AC 58 ms
63,360 KB
testcase_11 AC 549 ms
102,608 KB
testcase_12 AC 510 ms
99,296 KB
testcase_13 AC 420 ms
95,780 KB
testcase_14 AC 424 ms
96,544 KB
testcase_15 AC 622 ms
108,588 KB
testcase_16 AC 610 ms
103,088 KB
testcase_17 AC 592 ms
104,480 KB
testcase_18 AC 618 ms
106,572 KB
testcase_19 AC 568 ms
101,928 KB
testcase_20 AC 618 ms
106,020 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