結果
問題 | No.877 Range ReLU Query |
ユーザー | tktk_snsn |
提出日時 | 2020-09-29 00:02:40 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 724 ms / 2,000 ms |
コード長 | 1,715 bytes |
コンパイル時間 | 241 ms |
コンパイル使用メモリ | 82,220 KB |
実行使用メモリ | 102,784 KB |
最終ジャッジ日時 | 2024-04-25 22:26:13 |
合計ジャッジ時間 | 8,528 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 35 ms
53,424 KB |
testcase_01 | AC | 54 ms
66,584 KB |
testcase_02 | AC | 50 ms
64,588 KB |
testcase_03 | AC | 60 ms
71,068 KB |
testcase_04 | AC | 40 ms
59,992 KB |
testcase_05 | AC | 45 ms
62,156 KB |
testcase_06 | AC | 49 ms
62,664 KB |
testcase_07 | AC | 45 ms
62,320 KB |
testcase_08 | AC | 61 ms
72,776 KB |
testcase_09 | AC | 41 ms
60,092 KB |
testcase_10 | AC | 48 ms
64,400 KB |
testcase_11 | AC | 629 ms
101,392 KB |
testcase_12 | AC | 571 ms
98,552 KB |
testcase_13 | AC | 470 ms
94,152 KB |
testcase_14 | AC | 481 ms
95,776 KB |
testcase_15 | AC | 687 ms
102,444 KB |
testcase_16 | AC | 666 ms
100,940 KB |
testcase_17 | AC | 681 ms
102,108 KB |
testcase_18 | AC | 695 ms
101,264 KB |
testcase_19 | AC | 685 ms
100,260 KB |
testcase_20 | AC | 724 ms
102,784 KB |
ソースコード
import sys input = sys.stdin.buffer.readline sys.setrecursionlimit(10 ** 7) class fenwick_tree(object): def __init__(self, n): self.n = n self.log = n.bit_length() self.data = [0] * n def __sum(self, r): s = 0 while r > 0: s += self.data[r - 1] r -= r & -r return s def add(self, p, x): """ a[p] += xを行う""" p += 1 while p <= self.n: self.data[p - 1] += x p += p & -p def sum(self, l, r): """a[l] + a[l+1] + .. + a[r-1]を返す""" return self.__sum(r) - self.__sum(l) def lower_bound(self, x): """a[0] + a[1] + .. a[i] >= x となる最小のiを返す""" if x <= 0: return -1 i = 0 k = 1 << self.log while k: if i + k <= self.n and self.data[i + k - 1] < x: x -= self.data[i + k - 1] i += k k >>= 1 return i N, Q = map(int, input().split()) A = [(a, i) for i, a in enumerate(map(int, input().split()))] query = [] for i in range(Q): _, l, r, x = map(int, input().split()) query.append((x, l, r, i)) A.sort() query.sort() ans = [-1] * Q bit_A = fenwick_tree(N) bit_cnt = fenwick_tree(N) while A and query: if A[-1][0] >= query[-1][0]: a, i = A.pop() bit_A.add(i, a) bit_cnt.add(i, 1) else: x, l, r, i = query.pop() ans[i] = bit_A.sum(l - 1, r) - x * bit_cnt.sum(l - 1, r) while A: a, i = A.pop() bit_A.add(i, a) bit_cnt.add(i, 1) while query: x, l, r, i = query.pop() ans[i] = bit_A.sum(l - 1, r) - x * bit_cnt.sum(l - 1, r) print(*ans, sep="\n")