結果
問題 | No.1705 Mode of long array |
ユーザー |
![]() |
提出日時 | 2021-10-10 12:41:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 448 ms / 3,000 ms |
コード長 | 2,048 bytes |
コンパイル時間 | 220 ms |
コンパイル使用メモリ | 82,120 KB |
実行使用メモリ | 92,388 KB |
最終ジャッジ日時 | 2024-09-14 07:05:15 |
合計ジャッジ時間 | 15,533 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 51 |
ソースコード
import sys#input = sys.stdin.readlineinput = sys.stdin.buffer.readlineclass 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):self.update(i, 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)def main():N,M = map(int,input().split())A = list(map(int,input().split()))Tree = SegmentTree([0]*M,max,-1)for i in range(M):Tree.add(i,A[i]) #0-indexで追加していることに注意Q = int(input())for _ in range(Q):t,x,y = map(int,input().split())x -= 1 #0-indexif t == 1:Tree.add(x,y)if t == 2:Tree.add(x,-y)if t == 3:mx = Tree.sum(0,M)ok = Mng = 0while abs(ok-ng) > 1:mid = (ok+ng)//2if Tree.sum(mid,M) < mx: #[mid,M)にmxが入っていないok = midelse:ng = midprint(ok) #1indexでおkif __name__ == '__main__':main()