結果
問題 | No.1435 Mmm...... |
ユーザー | mkawa2 |
提出日時 | 2022-07-19 23:50:21 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 972 ms / 2,000 ms |
コード長 | 4,362 bytes |
コンパイル時間 | 173 ms |
コンパイル使用メモリ | 82,448 KB |
実行使用メモリ | 138,712 KB |
最終ジャッジ日時 | 2024-07-02 03:41:42 |
合計ジャッジ時間 | 15,945 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 76 ms
67,840 KB |
testcase_01 | AC | 73 ms
68,096 KB |
testcase_02 | AC | 77 ms
67,840 KB |
testcase_03 | AC | 72 ms
68,096 KB |
testcase_04 | AC | 72 ms
68,096 KB |
testcase_05 | AC | 73 ms
67,840 KB |
testcase_06 | AC | 527 ms
128,292 KB |
testcase_07 | AC | 554 ms
130,932 KB |
testcase_08 | AC | 577 ms
134,276 KB |
testcase_09 | AC | 607 ms
133,904 KB |
testcase_10 | AC | 561 ms
130,552 KB |
testcase_11 | AC | 554 ms
129,724 KB |
testcase_12 | AC | 546 ms
129,080 KB |
testcase_13 | AC | 547 ms
128,880 KB |
testcase_14 | AC | 464 ms
108,216 KB |
testcase_15 | AC | 605 ms
135,816 KB |
testcase_16 | AC | 595 ms
131,880 KB |
testcase_17 | AC | 450 ms
109,580 KB |
testcase_18 | AC | 654 ms
136,588 KB |
testcase_19 | AC | 575 ms
129,996 KB |
testcase_20 | AC | 636 ms
134,128 KB |
testcase_21 | AC | 563 ms
136,960 KB |
testcase_22 | AC | 972 ms
136,592 KB |
testcase_23 | AC | 634 ms
137,876 KB |
testcase_24 | AC | 622 ms
138,080 KB |
testcase_25 | AC | 619 ms
137,952 KB |
testcase_26 | AC | 679 ms
138,712 KB |
testcase_27 | AC | 643 ms
138,072 KB |
ソースコード
import sys sys.setrecursionlimit(200005) int1 = lambda x: int(x)-1 pDB = lambda *x: print(*x, end="\n", file=sys.stderr) p2D = lambda x: print(*x, sep="\n", end="\n\n", file=sys.stderr) def II(): return int(sys.stdin.readline()) def LI(): return list(map(int, sys.stdin.readline().split())) def LLI(rows_number): return [LI() for _ in range(rows_number)] def LI1(): return list(map(int1, sys.stdin.readline().split())) def LLI1(rows_number): return [LI1() for _ in range(rows_number)] def SI(): return sys.stdin.readline().rstrip() dij = [(0, 1), (-1, 0), (0, -1), (1, 0)] # dij = [(0, 1), (-1, 0), (0, -1), (1, 0), (1, 1), (1, -1), (-1, 1), (-1, -1)] inf = (1 << 63)-1 # inf = (1 << 31)-1 md = 10**9+7 # md = 998244353 import typing class SegTree: def __init__(self, op: typing.Callable[[typing.Any, typing.Any], typing.Any], e: typing.Any, v: typing.Union[int, typing.List[typing.Any]]) -> None: self._op = op self._e = e if isinstance(v, int): v = [e]*v self._n = len(v) self._log = (self._n-1).bit_length() self._size = 1 << self._log self._d = [e]*(2*self._size) for i in range(self._n): self._d[self._size+i] = v[i] for i in range(self._size-1, 0, -1): self._update(i) def set(self, p: int, x: typing.Any) -> None: assert 0 <= p < self._n p += self._size self._d[p] = x for i in range(1, self._log+1): self._update(p >> i) def get(self, p: int) -> typing.Any: assert 0 <= p < self._n return self._d[p+self._size] def prod(self, left: int, right: int) -> typing.Any: assert 0 <= left <= right <= self._n sml = self._e smr = self._e left += self._size right += self._size while left < right: if left & 1: sml = self._op(sml, self._d[left]) left += 1 if right & 1: right -= 1 smr = self._op(self._d[right], smr) left >>= 1 right >>= 1 return self._op(sml, smr) def all_prod(self) -> typing.Any: return self._d[1] def max_right(self, left: int, f: typing.Callable[[typing.Any], bool]) -> int: assert 0 <= left <= self._n assert f(self._e) if left == self._n: return self._n left += self._size sm = self._e first = True while first or (left & -left) != left: first = False while left%2 == 0: left >>= 1 if not f(self._op(sm, self._d[left])): while left < self._size: left *= 2 if f(self._op(sm, self._d[left])): sm = self._op(sm, self._d[left]) left += 1 return left-self._size sm = self._op(sm, self._d[left]) left += 1 return self._n def min_left(self, right: int, f: typing.Callable[[typing.Any], bool]) -> int: assert 0 <= right <= self._n assert f(self._e) if right == 0: return 0 right += self._size sm = self._e first = True while first or (right & -right) != right: first = False right -= 1 while right > 1 and right%2: right >>= 1 if not f(self._op(self._d[right], sm)): while right < self._size: right = 2*right+1 if f(self._op(self._d[right], sm)): sm = self._op(self._d[right], sm) right -= 1 return right+1-self._size sm = self._op(self._d[right], sm) return 0 def _update(self, k: int) -> None: self._d[k] = self._op(self._d[2*k], self._d[2*k+1]) def op(a, b): mm = [a[0], a[1], b[0], b[1]] mm.sort() return (mm[0], mm[1], max(a[2], b[2])) e = (inf, inf, -inf) n = II() aa = LI() st = SegTree(op, e, [(a, inf, a) for a in aa]) l = 0 ans = 0 for r in range(2, n+1): while l < r-1: m1, m2, M = st.prod(l, r) if m1+m2 >= M: break l += 1 ans += r-l-1 print(ans)