結果
問題 | No.1099 Range Square Sum |
ユーザー | tamato |
提出日時 | 2020-09-25 11:24:38 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 850 ms / 2,000 ms |
コード長 | 4,001 bytes |
コンパイル時間 | 1,618 ms |
コンパイル使用メモリ | 87,580 KB |
実行使用メモリ | 146,936 KB |
最終ジャッジ日時 | 2023-09-10 14:29:42 |
合計ジャッジ時間 | 11,232 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 72 ms
71,752 KB |
testcase_01 | AC | 70 ms
71,376 KB |
testcase_02 | AC | 76 ms
71,776 KB |
testcase_03 | AC | 70 ms
71,644 KB |
testcase_04 | AC | 72 ms
71,620 KB |
testcase_05 | AC | 70 ms
71,516 KB |
testcase_06 | AC | 73 ms
71,556 KB |
testcase_07 | AC | 72 ms
71,716 KB |
testcase_08 | AC | 71 ms
71,488 KB |
testcase_09 | AC | 73 ms
71,648 KB |
testcase_10 | AC | 72 ms
71,760 KB |
testcase_11 | AC | 141 ms
78,292 KB |
testcase_12 | AC | 139 ms
79,120 KB |
testcase_13 | AC | 130 ms
78,336 KB |
testcase_14 | AC | 130 ms
77,952 KB |
testcase_15 | AC | 133 ms
78,348 KB |
testcase_16 | AC | 137 ms
78,116 KB |
testcase_17 | AC | 137 ms
78,888 KB |
testcase_18 | AC | 133 ms
78,316 KB |
testcase_19 | AC | 132 ms
78,292 KB |
testcase_20 | AC | 131 ms
78,048 KB |
testcase_21 | AC | 834 ms
146,936 KB |
testcase_22 | AC | 850 ms
145,112 KB |
testcase_23 | AC | 837 ms
146,484 KB |
testcase_24 | AC | 838 ms
146,272 KB |
testcase_25 | AC | 834 ms
145,168 KB |
testcase_26 | AC | 503 ms
113,876 KB |
testcase_27 | AC | 505 ms
113,740 KB |
testcase_28 | AC | 517 ms
113,820 KB |
testcase_29 | AC | 501 ms
114,212 KB |
testcase_30 | AC | 508 ms
113,824 KB |
ソースコード
mod = 1000000007 eps = 10**-9 mask = (1 << 60) - 1 def main(): import sys input = sys.stdin.buffer.readline def op(a, b): return (a[0] + b[0], a[1] + b[1]) e = (0, 0) def mapping(a, x, seg_len): sqsum, sum_ = a sqsum_new = sqsum + seg_len * (x ** 2) + 2 * x * sum_ sum_new = sum_ + seg_len * x return (sqsum_new, sum_new) def composition(x, y): return x + y id = 0 class LazySegTree: # Range update query def __init__(self, A, op=op, e=e, mapping=mapping, composition=composition, id=id, initialize=True): self.N = len(A) self.LV = (self.N - 1).bit_length() self.N0 = 1 << self.LV self.op = op self.e = e self.mapping = mapping self.composition = composition self.id = id if initialize: self.data = [self.e] * self.N0 + A + [self.e] * (self.N0 - self.N) for i in range(self.N0 - 1, 0, -1): self.data[i] = op(self.data[i * 2], self.data[i * 2 + 1]) else: self.data = [self.e] * (self.N0 * 2) self.lazy = [id] * (self.N0 * 2) def _ascend(self, i): for _ in range(i.bit_length() - 1): i >>= 1 self.data[i] = self.op(self.data[i * 2], self.data[i * 2 + 1]) def _descend(self, idx): lv = idx.bit_length() seg_len = 1 << self.LV for j in range(lv - 1, 0, -1): seg_len >>= 1 i = idx >> j x = self.lazy[i] if x == self.id: continue self.lazy[i * 2] = self.composition(self.lazy[i * 2], x) self.lazy[i * 2 + 1] = self.composition(self.lazy[i * 2 + 1], x) self.lazy[i] = self.id self.data[i * 2] = self.mapping(self.data[i * 2], x, seg_len) self.data[i * 2 + 1] = self.mapping(self.data[i * 2 + 1], x, seg_len) # open interval [l, r) def apply(self, l, r, x): l += self.N0 - 1 r += self.N0 - 1 self._descend(l // (l & -l)) self._descend(r // (r & -r) - 1) l_ori = l r_ori = r seg_len = 1 while l < r: if l & 1: self.data[l] = self.mapping(self.data[l], x, seg_len) self.lazy[l] = self.composition(self.lazy[l], x) l += 1 if r & 1: r -= 1 self.data[r] = self.mapping(self.data[r], x, seg_len) self.lazy[r] = self.composition(self.lazy[r], x) l >>= 1 r >>= 1 seg_len <<= 1 self._ascend(l_ori // (l_ori & -l_ori)) self._ascend(r_ori // (r_ori & -r_ori) - 1) # open interval [l, r) def query(self, l, r): l += self.N0 - 1 r += self.N0 - 1 self._descend(l // (l & -l)) self._descend(r // (r & -r) - 1) ret_l = self.e ret_r = self.e while l < r: if l & 1: ret_l = self.op(ret_l, self.data[l]) l += 1 if r & 1: ret_r = self.op(self.data[r - 1], ret_r) r -= 1 l >>= 1 r >>= 1 return self.op(ret_l, ret_r) N = int(input()) A = list(map(int, input().split())) B = [(a*a, a) for a in A] ST = LazySegTree(B) ans = [] for _ in range(int(input())): query = list(map(int, input().split())) if query[0] == 1: l, r, x = query[1:] ST.apply(l, r+1, x) else: l, r = query[1:] ans.append(ST.query(l, r+1)[0]) print(*ans, sep="\n") if __name__ == '__main__': main()