結果
問題 | No.1868 Teleporting Cyanmond |
ユーザー |
![]() |
提出日時 | 2022-03-11 21:42:41 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 380 ms / 2,000 ms |
コード長 | 3,494 bytes |
コンパイル時間 | 256 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 20,844 KB |
最終ジャッジ日時 | 2024-09-16 01:54:37 |
合計ジャッジ時間 | 4,998 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 |
ソースコード
class SegTree(object):def __init__(self, N, op_data, u_data):self._n = Nself.log = (N-1).bit_length()self.size = 1 << self.logself.op = op_dataself.e = u_dataself.data = [u_data] * (2 * self.size)# self.len = [1] * (2 * self.size)def __repr__(self):res = [self.get(i) for i in range(self._n)]return " ".join(map(str, res))def _update(self, i):self.data[i] = self.op(self.data[i << 1], self.data[i << 1 | 1])def initialize(self, arr=None):""" segtreeをarrで初期化する。len(arr) == Nにすること """if arr:for i, a in enumerate(arr, self.size):self.data[i] = afor i in reversed(range(1, self.size)):self._update(i)# self.len[i] = self.len[i << 1] + self.len[i << 1 | 1]def update(self, p, x):""" data[p] = x とする (0-indexed)"""p += self.sizeself.data[p] = xfor i in range(1, self.log + 1):self._update(p >> i)def get(self, p):""" data[p]を返す """return self.data[p + self.size]def prod(self, l, r):"""op_data(data[l], data[l+1], ..., data[r-1])を返す (0-indexed)"""sml = self.esmr = self.el += self.sizer += self.sizewhile l < r:if l & 1:sml = self.op(sml, self.data[l])l += 1if r & 1:r -= 1smr = self.op(self.data[r], smr)l >>= 1r >>= 1return self.op(sml, smr)def all_prod(self):""" op(data[0], data[1], ... data[N-1])を返す """return self.data[1]def max_right(self, l, func):"""func(l, l+1, ..., r-1) = True,func(l, l+1, ..., r-1, r) = Falseとなる r を返す"""if l == self._n:return self._nl += self.sizesm = self.ewhile True:while l % 2 == 0:l >>= 1if not func(self.op(sm, self.data[l])):while l < self.size:l <<= 1if func(self.op(sm, self.data[l])):sm = self.op(sm, self.data[l])l += 1return l - self.sizesm = self.op(sm, self.data[l])l += 1if (l & -l) == l:breakreturn self._ndef min_left(self, r, func):"""func( l, l+1, ..., r-1) = True,func(l-1, l, l+1, ..., r-1) = Falseとなる l を返す"""if r == 0:return 0r += self.sizesm = self.ewhile True:r -= 1while r > 1 and r & 1:r >>= 1if not func(self.op(self.data[r], sm)):while r < self.size:r = r << 1 | 1if func(self.op(self.data[r], sm)):sm = self.op(self.data[r], sm)r -= 1return r + 1 - self.sizesm = self.op(self.data[r], sm)if (r & -r) == r:breakreturn 0N = int(input())A = list(map(int, input().split()))seg = SegTree(N, max, 0)seg.initialize(A)ans = 0now = 1while now < N:now = seg.prod(0, now)ans += 1print(ans)