結果
問題 | No.2101 [Cherry Alpha N] ずっとこの数列だったらいいのに |
ユーザー |
![]() |
提出日時 | 2022-10-17 16:33:18 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,924 ms / 6,000 ms |
コード長 | 2,240 bytes |
コンパイル時間 | 691 ms |
コンパイル使用メモリ | 82,352 KB |
実行使用メモリ | 249,344 KB |
最終ジャッジ日時 | 2024-06-28 06:56:12 |
合計ジャッジ時間 | 87,598 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 42 |
ソースコード
class SegmentTree(object):def __init__(self, A, dot, unit):n = 1 << (len(A) - 1).bit_length()tree = [unit] * (2 * n)for i, v in enumerate(A):tree[i + n] = vfor i in range(n - 1, 0, -1):tree[i] = dot(tree[i << 1], tree[i << 1 | 1])self._n = nself._tree = treeself._dot = dotself._unit = unitdef __getitem__(self, i):return self._tree[i + self._n]def update(self, i, v):i += self._nself._tree[i] = vwhile i != 1:i >>= 1self._tree[i] = self._dot(self._tree[i << 1], self._tree[i << 1 | 1])def add(self, i, v,ope):self.update(i, ope(self[i],v))def sum(self, l, r): #これで[l,r)から取り出す。l += self._nr += self._nl_val = r_val = self._unitwhile l < r:if l & 1:l_val = self._dot(l_val, self._tree[l])l += 1if r & 1:r -= 1r_val = self._dot(self._tree[r], r_val)l >>= 1r >>= 1return self._dot(l_val, r_val)from collections import defaultdictN = int(input())A = []T = []for _ in range(N):a,t = map(int,input().split())A.append(a)T.append(t)Q = int(input())memo = defaultdict(list)board = []for i in range(N):memo[T[i]].append((-1,i))memo[T[i] + A[i]].append((0,i))for q in range(Q):d,l,r = map(int,input().split())l-=1r-=1memo[d].append((1,q))board.append((l,r))ANS = [0] * Qkeys = list(memo.keys())keys.sort()segcnt = SegmentTree([0] * N,lambda x,y:x+y,0)segsum = SegmentTree([0] * N ,lambda x,y:x+y,0)Aseg = SegmentTree(A,lambda x,y:x+y,0)for key in keys:memo[key].sort()for t,index in memo[key]:if t == -1:segcnt.update(index,1)segsum.update(index,-(key-1))elif t == 0:segcnt.update(index,0)segsum.update(index,0)Aseg.update(index,0)else:l,r = board[index]ans = Aseg.sum(l,r+1) - (segcnt.sum(l,r+1) * key + segsum.sum(l,r+1))ANS[index] = ansprint(*ANS,sep='\n')