#https://github.com/shakayami/ACL-for-python/blob/master/lazysegtree.py class lazy_segtree(): def update(self,k):self.d[k]=self.op(self.d[2*k],self.d[2*k+1]) def all_apply(self,k,f): self.d[k]=self.mapping(f,self.d[k]) if (k>i) self.d[p]=x for i in range(1,self.log+1):self.update(p>>i) def get(self,p): assert 0<=p and p>i) return self.d[p] def prod(self,l,r): assert 0<=l and l<=r and r<=self.n if l==r:return self.e l+=self.size r+=self.size for i in range(self.log,0,-1): if (((l>>i)<>i) if (((r>>i)<>i) sml,smr=self.e,self.e while(l>=1 r>>=1 return self.op(sml,smr) def all_prod(self):return self.d[1] def apply_point(self,p,f): assert 0<=p and p>i) self.d[p]=self.mapping(f,self.d[p]) for i in range(1,self.log+1):self.update(p>>i) def apply(self,l,r,f): assert 0<=l and l<=r and r<=self.n if l==r:return l+=self.size r+=self.size for i in range(self.log,0,-1): if (((l>>i)<>i) if (((r>>i)<>i) l2,r2=l,r while(l>=1 r>>=1 l,r=l2,r2 for i in range(1,self.log+1): if (((l>>i)<>i) if (((r>>i)<>i) def max_right(self,l,g): assert 0<=l and l<=self.n assert g(self.e) if l==self.n:return self.n l+=self.size for i in range(self.log,0,-1):self.push(l>>i) sm=self.e while(1): while(l%2==0):l>>=1 if not(g(self.op(sm,self.d[l]))): while(l>i) sm=self.e while(1): r-=1 while(r>1 and (r%2)):r>>=1 if not(g(self.op(self.d[r],sm))): while(r= 0: ET.append(i) ct += 1 if ET1[i] == 0: ET1[i] = ct de += 1 DE[i] = de for a in self.edge[i][::-1]: if a != P[i]: P[a] = i for k in range(len(self.edge[a])): if self.edge[a][k] == i: del self.edge[a][k] break Q.append(~a) Q.append(a) self.et = ET self.et1 = ET1 self.et2 = ET2 self.et_sorted = list(sorted(self.et1)) self.l = list(range(n)) self.l.sort(key=lambda x:self.et1[x]) def subtree(self,x): left,right = self.et1[x],self.et2[x] left = bisect.bisect_left(self.et_sorted,left) right = bisect.bisect_left(self.et_sorted,right) return left,right import collections,heapq start = 0 d = collections.deque() dist = [10**18 for i in range(n1)] d.append(start) dist[start] = 0 dist2 = [0 for i in range(n1)] p = [-1 for i in range(n1)] while d: now = d.popleft() for i in edge1[now]: if dist[i] > dist[now] + 1: dist[i] = dist[now] + 1 d.append(i) p[i] = now dist2[i] = dist2[now] + cc[i*n1+now] import operator v1 = Vertex_Add_Subtree_Sum(n1,edge1) lst1 = lazy_segtree([dist2[i] for i in v1.l],max,-10**18,operator.add,operator.add,0) a = [-1 for i in range(n1)] l0,r0 = v1.subtree(0) #print(v1.et) for i in v1.et: if i >= 0: if i != 0: cost = cc[p[i]*n1+i] l,r = v1.subtree(i) lst1.apply(l,r,-2 * cost) lst1.apply(l0,r0,1 * cost) a[i] = lst1.all_prod() if i < 0: j = ~i cost = cc[p[j]*n1+j] l,r = v1.subtree(j) lst1.apply(l,r,2 * cost) lst1.apply(l0,r0,-1 * cost) print(max(a))