結果

問題 No.877 Range ReLU Query
ユーザー H3PO4H3PO4
提出日時 2021-03-04 11:55:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 676 ms / 2,000 ms
コード長 1,043 bytes
コンパイル時間 410 ms
コンパイル使用メモリ 86,720 KB
実行使用メモリ 107,996 KB
最終ジャッジ日時 2023-08-08 04:49:43
合計ジャッジ時間 9,279 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 76 ms
70,876 KB
testcase_01 AC 90 ms
75,624 KB
testcase_02 AC 90 ms
75,596 KB
testcase_03 AC 96 ms
75,980 KB
testcase_04 AC 80 ms
75,584 KB
testcase_05 AC 85 ms
75,328 KB
testcase_06 AC 84 ms
75,408 KB
testcase_07 AC 84 ms
75,652 KB
testcase_08 AC 91 ms
75,732 KB
testcase_09 AC 80 ms
75,556 KB
testcase_10 AC 89 ms
75,696 KB
testcase_11 AC 618 ms
103,804 KB
testcase_12 AC 566 ms
101,368 KB
testcase_13 AC 469 ms
96,648 KB
testcase_14 AC 474 ms
97,388 KB
testcase_15 AC 667 ms
107,996 KB
testcase_16 AC 661 ms
104,288 KB
testcase_17 AC 675 ms
106,996 KB
testcase_18 AC 665 ms
106,236 KB
testcase_19 AC 623 ms
102,860 KB
testcase_20 AC 676 ms
106,704 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