結果
問題 | No.2942 Sigma Music Game Level Problem |
ユーザー |
|
提出日時 | 2024-10-18 23:26:43 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 3,410 ms / 6,000 ms |
コード長 | 2,244 bytes |
コンパイル時間 | 1,212 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 292,188 KB |
最終ジャッジ日時 | 2024-11-15 06:57:25 |
合計ジャッジ時間 | 31,714 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
class SegTree:def __init__(self, op, e, n, v=None):self._n = nself._op = opself._e = eself._log = (n - 1).bit_length()self._size = 1 << self._logself._d = [self._e()] * (self._size << 1)if v is not None:for i in range(self._n):self._d[self._size + i] = v[i]for i in range(self._size - 1, 0, -1):self._d[i] = self._op(self._d[i << 1], self._d[i << 1 | 1])def set(self, p, x):p += self._sizeself._d[p] = xwhile p:l, r = p, p^1if l > r: l, r = r, lself._d[p >> 1] = self._op(self._d[l], self._d[r])p >>= 1def get(self, p):return self._d[p + self._size]#[l, r)の区間で求めるdef prod(self, l, r):sml, smr = self._e(), self._e()l += self._sizer += self._sizewhile l < r:if l & 1:sml = self._op(sml, self._d[l])l += 1if r & 1:r -= 1smr = self._op(self._d[r], smr)l >>= 1r >>= 1return self._op(sml, smr)def all_prod(self):return self._d[1]def op(x, y):return x+ydef e():return 0BASE = 21MASK = (1<<BASE)-1def op2(x, y):xi, xc = x>>BASE, x&MASKyi, yc = y>>BASE, y&MASKret = ((xi+yi)<<BASE) | (xc+yc)return retdef e2():return 0import sysinput = sys.stdin.readlineN, Q, L = map(int, input().split())A = list(map(int, input().split()))query = [list(map(int, input().split())) for _ in range(Q)]dp = [0 for _ in range(200001)]for a in A:dp[a] += 1for i in range(200001):c = dp[i]dp[i] = ((c*i)<<BASE) | cST = SegTree(op2, e2, 200001, dp)flag = Truefor q in query:t = q[0]if t==1:l = q[1]v = ST.get(l)vi, vc = v>>BASE, v&MASKvi += lvc += 1v = (vi<<BASE)|vcST.set(l, v)elif t==2:l, r = q[1:]v = ST.prod(l, r+1)vi, vc = v>>BASE, v&MASKprint(vc, vi)flag = Falseelse:m = q[1]L = mif flag:print("Not Found!")