結果
問題 | No.1096 Range Sums |
ユーザー | c-yan |
提出日時 | 2020-06-26 22:22:22 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 876 ms / 2,000 ms |
コード長 | 1,164 bytes |
コンパイル時間 | 139 ms |
コンパイル使用メモリ | 10,964 KB |
実行使用メモリ | 31,284 KB |
最終ジャッジ日時 | 2023-09-18 05:42:01 |
合計ジャッジ時間 | 5,450 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 17 ms
8,376 KB |
testcase_01 | AC | 16 ms
8,400 KB |
testcase_02 | AC | 16 ms
8,164 KB |
testcase_03 | AC | 16 ms
8,248 KB |
testcase_04 | AC | 16 ms
8,220 KB |
testcase_05 | AC | 16 ms
8,208 KB |
testcase_06 | AC | 17 ms
8,384 KB |
testcase_07 | AC | 16 ms
8,220 KB |
testcase_08 | AC | 16 ms
8,408 KB |
testcase_09 | AC | 16 ms
8,316 KB |
testcase_10 | AC | 869 ms
31,244 KB |
testcase_11 | AC | 868 ms
31,152 KB |
testcase_12 | AC | 867 ms
31,248 KB |
testcase_13 | AC | 874 ms
31,284 KB |
testcase_14 | AC | 876 ms
31,132 KB |
ソースコード
from itertools import accumulate class SegmentTree(): _data = [] _offset = 0 _size = 0 def __init__(self, size): _size = size t = 1 while t < size: t *= 2 self._offset = t - 1 self._data = [0 for _ in range(t * 2 - 1)] def update_all(self, iterable): data = self._data data[self._offset:self._offset + self._size] = iterable for i in range(self._offset - 1, -1, -1): data[i] = data[i * 2 + 1] + data[i * 2 + 2] def query(self, start, stop): result = 0 l = start + self._offset r = stop + self._offset while l < r: if l & 1 == 0: result = result + self._data[l] if r & 1 == 0: result = result + self._data[r - 1] l = l // 2 r = (r - 1) // 2 return result N, *A = map(int, open(0).read().split()) a = list(accumulate(A)) st = SegmentTree(N) st.update_all(a) #print(a) result = 0 result += st.query(0, N) #print(result) for i in range(1, N): result += st.query(i, N) - a[i - 1] * (N - i) #print(result) print(result)