結果

問題 No.1031 いたずら好きなお姉ちゃん
ユーザー shotoyooshotoyoo
提出日時 2021-05-17 23:05:47
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 5,024 bytes
コンパイル時間 382 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 101,504 KB
最終ジャッジ日時 2024-10-07 04:03:52
合計ジャッジ時間 9,635 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 RE -
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 RE -
testcase_45 RE -
testcase_46 RE -
testcase_47 RE -
testcase_48 RE -
testcase_49 RE -
testcase_50 RE -
testcase_51 RE -
testcase_52 RE -
testcase_53 RE -
testcase_54 RE -
testcase_55 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()

sys.setrecursionlimit(2*10**5+10)
write = lambda x: sys.stdout.write(x+"\n")
debug = lambda x: sys.stderr.write(x+"\n")
writef = lambda x: print("{:.12f}".format(x))

### セグメント木(はやい)
class SG:
    def __init__(self, n, v=None, ninf=10**12):
        self._n = n
        self.ninf = ninf
        x = 0
        while (1 << x) < n:
            x += 1
        self._log = x
        self._size = 1 << self._log
        self._d = [self.ninf] * (2 * self._size)
        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._update(i)
    def _update(self, k):
        self._d[k] = op(self._d[2 * k], self._d[2 * k + 1])
    def update(self, p, x):
        assert 0 <= p < self._n
        p += self._size
        self._d[p] = x
        for i in range(1, self._log + 1):
#             self._update(p >> i)
            k = p>>i
            self._d[k] = op(self._d[2 * k], self._d[2 * k + 1])
    def get(self, p):
#         assert 0 <= p < self._n
        return self._d[p + self._size]
    def check(self):
        return [self.get(p) for p in range(self._n)]
    def query(self, left, right):
        # [l,r)の総和
        assert 0 <= left <= right <= self._n
        sml = self.ninf
        smr = self.ninf
        left += self._size
        right += self._size
        # 外側から計算していく(lは小さい側から, rは大きい側から)
        while left < right:
            if left & 1:
                sml = op(sml, self._d[left])
                left += 1
            if right & 1:
                right -= 1
                smr = op(self._d[right], smr)
            left >>= 1
            right >>= 1
        return op(sml, smr)
    def query_all(self):
        return self._d[1]
    def max_right(self, left, f):
        """f(op(a[l], a[l + 1], ..., a[r - 1])) = true となる最大の r
        -> rはf(op(a[l:r+1]))がFalseになる最小のr
        """
#         assert 0 <= left <= self._n
#         assert f(ninf)
        if left == self._n:
            return self._n
        left += self._size
        sm = self.ninf
        first = True
        while first or (left & -left) != left:
            first = False
            while left % 2 == 0:
                left >>= 1
            if not f(op(sm, self._d[left])):
                while left < self._size:
                    left *= 2
                    if f(op(sm, self._d[left])):
                        sm = op(sm, self._d[left])
                        left += 1
                return left - self._size
            sm = op(sm, self._d[left])
            left += 1
        return self._n
    def min_left(self, right, f):
        """f(op(a[l], a[l + 1], ..., a[r - 1])) = true となる最小の l
        """
#         assert 0 <= right <= self._n
#         assert f(ninf)
        if right == 0:
            return 0
        right += self._size
        sm = self.ninf
        first = True
        while first or (right & -right) != right:
            first = False
            right -= 1
            while right > 1 and right % 2:
                right >>= 1
            if not f(op(self._d[right], sm)):
                while right < self._size:
                    right = 2 * right + 1
                    if f(op(self._d[right], sm)):
                        sm = op(self._d[right], sm)
                        right -= 1
                return right + 1 - self._size
            sm = op(self._d[right], sm)
        return 0
def nearest_larger(a):
    """a[i]よりa[j]が大きいようなjの最小値
    """
    s = [inf]
    si = [None]
    n = len(a)
    ans = [-1]*n
    for i in range(n)[::-1]:
        v = a[i]
        while s[-1]<v:
            s.pop()
            si.pop()
        ans[i] = si[-1]
        s.append(v)
        si.append(i)
    return ans
class L2RM:
    """left-to-right maxima
    配列aをi->jにいくときに、最大値が何回更新されるか
    """
    def __init__(self, a):
        self.n = len(a)
        s = [inf]
        si = [-1]
        n = len(a)
        ds = [0]*n
        for i in range(n)[::-1]:
            v = a[i]
            while s[-1]<v:
                s.pop()
                si.pop()
            if si[-1]!=-1:
                ds[i] = ds[si[-1]] + 1
            s.append(v)
            si.append(i)
        self.sg = SG(n, ds, ninf=10**12)
    def query(self, l, r):
        """[l,r]で最大値が何回更新されるか
        """
        return self.sg.get(l) - self.sg.query(l,r+1)
op = min

n = int(input())
a = list(map(int, input().split()))
sg = SG(n,[(v,i) for i,v in enumerate(a)],ninf=(10**12,-1))
l1 = L2RM(a)
l2 = L2RM(a[::-1])
q = [(0,n-1)]
ans = 0
while q:
    l,r = q.pop()
    if l==r:
        continue
    v,m = sg.query(l,r+1)
    ans += l1.query(m,r) + l2.query(n-1-m, n-1-l)
    if l!=m:
        q.append((l,m-1))
    if r!=m:
        q.append((m+1,r))
print(ans)
0