結果

問題 No.1099 Range Square Sum
ユーザー convexineq
提出日時 2021-05-11 06:25:12
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 6,245 bytes
コンパイル時間 298 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 281,788 KB
最終ジャッジ日時 2024-09-21 12:08:08
合計ジャッジ時間 21,200 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 25 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

class LazySegmentTree:
#seg = LazySegmentTree(op_X, e_X, mapping, composision_of_Aut_X, id_of_Aut_X, N, array=None)
def __init__(self, op_X, e_X, mapping, composision_of_Aut_X, id_of_Aut_X, N, array=None):
# X, , f(x), f\circ g, X
# M X
#__slots__ = ["op_X", "e_X", "mapping","compose","id_M","N","log","N0","data","lazy"]
self.e_X = e_X; self.op_X = op_X; self.mapping = mapping; self.compose = composision_of_Aut_X; self.id_M = id_of_Aut_X
self.N = N
self.log = (N-1).bit_length()
self.N0 = 1<<self.log
self.data = [e_X]*(2*self.N0)
self.lazy = [self.id_M]*self.N0
if array is not None:
assert N == len(array)
self.data[self.N0:self.N0+self.N] = array
for i in range(self.N0-1,0,-1): self.update(i)
# 1
def point_set(self, p, x):
p += self.N0
for i in range(self.log, 0,-1):
self.push(p>>i)
self.data[p] = x
for i in range(1, self.log + 1):
self.update(p>>i)
# 1
def point_get(self, p):
p += self.N0
for i in range(self.log, 0, -1):
self.push(p>>i)
return self.data[p]
# [L,R)op
def prod(self, l, r):
if l == r: return self.e_X
l += self.N0
r += self.N0
for i in range(self.log, 0, -1):
if (l>>i)<<i != l:
self.push(l>>i)
if (r>>i)<<i != r:
self.push(r>>i)
sml = smr = self.e_X
while l < r:
if l & 1:
sml = self.op_X(sml, self.data[l])
l += 1
if r & 1:
r -= 1
smr = self.op_X(self.data[r], smr)
l >>= 1
r >>= 1
return self.op_X(sml, smr)
# op
def all_prod(self): return self.data[1]
# 1
def apply_point(self, p, f):
p += self.N0
for i in range(self.log, 0, -1):
self.push(p>>i)
self.data[p] = self.mapping(f, self.data[p])
for i in range(1, self.log + 1):
self.update(p>>i)
#
def apply(self, l, r, f):
if l == r: return
l += self.N0
r += self.N0
for i in range(self.log, 0, -1):
if (l>>i)<<i != l:
self.push(l>>i)
if (r>>i)<<i != r:
self.push((r-1)>>i)
l2, r2 = l, r
while l < r:
if l & 1:
self.all_apply(l, f)
l += 1
if r & 1:
r -= 1
self.all_apply(r, f)
l >>= 1
r >>= 1
l, r = l2, r2
for i in range(1, self.log + 1):
if (l>>i)<<i != l:
self.update(l>>i)
if (r>>i)<<i != r:
self.update((r-1)>>i)
"""
l
f(x_l*...*x_{r-1}) True r
TTTTFFFF F
n
f(e_M) = True
"""
def max_right(self, l, g):
if l == self.N: return self.N
l += self.N0
for i in range(self.log, 0, -1): self.push(l>>i)
sm = self.e_X
while True:
while l&1 == 0:
l >>= 1
if not g(self.op_X(sm, self.data[l])):
while l < self.N0:
self.push(l)
l *= 2
if g(self.op_X(sm, self.data[l])):
sm = self.op_X(sm, self.data[l])
l += 1
return l - self.N0
sm = self.op_X(sm, self.data[l])
l += 1
if l&-l == l: break
return self.N
"""
r
f(x_l*...*x_{r-1}) True l
FFFFTTTT T
r
f(e_M) = True
"""
def min_left(self, r, g):
if r == 0: return 0
r += self.N0
for i in range(self.log, 0, -1): self.push((r-1)>>i)
sm = self.e_X
while True:
r -= 1
while r>1 and r&1:
r >>= 1
if not g(self.op_X(self.data[r], sm)):
while r < self.N0:
self.push(r)
r = 2*r + 1
if g(self.op_X(self.data[r], sm)):
sm = self.op_X(self.data[r], sm)
r -= 1
return r + 1 - self.N0
sm = self.op_X(self.data[r], sm)
if r&-r == r: break
return 0
#
def update(self, k):
self.data[k] = self.op_X(self.data[2*k], self.data[2*k+1])
def all_apply(self, k, f):
self.data[k] = self.mapping(f, self.data[k])
if k < self.N0:
self.lazy[k] = self.compose(f, self.lazy[k])
def push(self, k): #propagate
if self.lazy[k] is self.id_M: return
self.data[2*k ] = self.mapping(self.lazy[k], self.data[2*k])
self.data[2*k+1] = self.mapping(self.lazy[k], self.data[2*k+1])
if 2*k < self.N0:
self.lazy[2*k] = self.compose(self.lazy[k], self.lazy[2*k])
self.lazy[2*k+1] = self.compose(self.lazy[k], self.lazy[2*k+1])
self.lazy[k] = self.id_M
def op_X(X,Y):
return(X[0]+Y[0],X[1]+Y[1],X[2]+Y[2])
e_X = (0,0,0)
def mapping(v,X):
a,b,d = X
return (a+v*d, b+v*(2*a+v*d), d)
def composision_of_Aut_X(v,w):
return v+w
id_of_Aut_X = 0
import sys
readline = sys.stdin.readline
n = int(readline())
*a, = map(int,readline().split())
array = [(ai,ai*ai,1) for ai in a]
seg = LazySegmentTree(op_X, e_X, mapping, composision_of_Aut_X, id_of_Aut_X, n, array)
Q = int(readline())
for _ in range(Q):
v,*q = map(int,readline().split())
if v==1:
l,r,x = q
seg.apply(l-1,r,x)
else:
l,r = q
print(seg.prod(l-1,r)[1])
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0