結果

問題 No.877 Range ReLU Query
コンテスト
ユーザー H3PO4
提出日時 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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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