結果

問題 No.1095 Smallest Kadomatsu Subsequence
ユーザー toyuzukotoyuzuko
提出日時 2020-06-27 14:27:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,214 ms / 2,000 ms
コード長 1,658 bytes
コンパイル時間 173 ms
コンパイル使用メモリ 82,068 KB
実行使用メモリ 108,952 KB
最終ジャッジ日時 2024-07-05 09:08:35
合計ジャッジ時間 13,001 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
52,608 KB
testcase_01 AC 37 ms
52,480 KB
testcase_02 AC 38 ms
52,224 KB
testcase_03 AC 46 ms
61,696 KB
testcase_04 AC 47 ms
60,800 KB
testcase_05 AC 47 ms
61,056 KB
testcase_06 AC 45 ms
61,056 KB
testcase_07 AC 47 ms
60,800 KB
testcase_08 AC 48 ms
61,056 KB
testcase_09 AC 46 ms
60,928 KB
testcase_10 AC 47 ms
61,104 KB
testcase_11 AC 47 ms
60,800 KB
testcase_12 AC 46 ms
61,440 KB
testcase_13 AC 154 ms
78,232 KB
testcase_14 AC 151 ms
78,112 KB
testcase_15 AC 157 ms
78,084 KB
testcase_16 AC 147 ms
78,272 KB
testcase_17 AC 148 ms
77,972 KB
testcase_18 AC 149 ms
77,968 KB
testcase_19 AC 153 ms
78,360 KB
testcase_20 AC 162 ms
78,376 KB
testcase_21 AC 150 ms
77,960 KB
testcase_22 AC 151 ms
78,016 KB
testcase_23 AC 1,190 ms
108,800 KB
testcase_24 AC 1,182 ms
108,416 KB
testcase_25 AC 1,190 ms
108,592 KB
testcase_26 AC 1,214 ms
108,824 KB
testcase_27 AC 1,214 ms
108,416 KB
testcase_28 AC 482 ms
108,952 KB
testcase_29 AC 567 ms
108,476 KB
testcase_30 AC 831 ms
108,404 KB
testcase_31 AC 852 ms
108,700 KB
testcase_32 AC 845 ms
108,788 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

class SegmentTree():
    def __init__(self, arr, func=min, ie=2**63):
        self.h = (len(arr) - 1).bit_length()
        self.n = 2**self.h
        self.ie = ie
        self.func = func
        self.tree = [ie for _ in range(2 * self.n)]
        for i in range(len(arr)):
            self.tree[self.n + i] = arr[i]
        for i in range(1, self.n)[::-1]:
            self.tree[i] = func(self.tree[2 * i], self.tree[2 * i + 1])

    def set(self, idx, x):
        idx += self.n
        self.tree[idx] = x
        while idx:
            idx >>= 1
            self.tree[idx] = self.func(self.tree[2 * idx], self.tree[2 * idx + 1])

    def query(self, lt, rt):
        lt += self.n
        rt += self.n
        vl = vr = self.ie
        while rt - lt > 0:
            if lt & 1:
                vl = self.func(vl, self.tree[lt])
                lt += 1
            if rt & 1:
                rt -= 1
                vr = self.func(self.tree[rt], vr)
            lt >>= 1
            rt >>= 1
        return self.func(vl, vr)

INF = 10**18

N = int(input())
A = list(map(int, input().split()))

S = sorted([(A[i], i) for i in range(N)])

res = INF

st1 = SegmentTree([INF] * N, min, INF)

for i in range(N):
    a, idx = S[i]
    st1.set(idx, a)
    lt = st1.query(0, idx)
    rt = st1.query(idx + 1, N)
    if lt == INF or rt == INF:
        continue
    res = min(res, lt + rt + a)

st2 = SegmentTree([INF] * N, min, INF)

for i in range(N)[::-1]:
    a, idx = S[i]
    st2.set(idx, a)
    lt = st2.query(0, idx)
    rt = st2.query(idx + 1, N)
    if lt == INF or rt == INF:
        continue
    res = min(res, lt + rt + a)

print(res if res != INF else -1)
0