結果

問題 No.1095 Smallest Kadomatsu Subsequence
ユーザー convexineqconvexineq
提出日時 2020-06-26 22:36:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 981 ms / 2,000 ms
コード長 3,312 bytes
コンパイル時間 312 ms
コンパイル使用メモリ 82,332 KB
実行使用メモリ 144,384 KB
最終ジャッジ日時 2024-07-04 22:23:11
合計ジャッジ時間 11,264 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
53,108 KB
testcase_01 AC 37 ms
53,292 KB
testcase_02 AC 38 ms
53,420 KB
testcase_03 AC 44 ms
61,388 KB
testcase_04 AC 46 ms
60,764 KB
testcase_05 AC 47 ms
61,428 KB
testcase_06 AC 47 ms
62,108 KB
testcase_07 AC 47 ms
61,112 KB
testcase_08 AC 48 ms
61,084 KB
testcase_09 AC 47 ms
62,296 KB
testcase_10 AC 47 ms
60,576 KB
testcase_11 AC 46 ms
60,456 KB
testcase_12 AC 45 ms
61,176 KB
testcase_13 AC 128 ms
78,740 KB
testcase_14 AC 137 ms
79,352 KB
testcase_15 AC 127 ms
78,660 KB
testcase_16 AC 129 ms
78,376 KB
testcase_17 AC 126 ms
78,448 KB
testcase_18 AC 133 ms
78,772 KB
testcase_19 AC 126 ms
78,336 KB
testcase_20 AC 126 ms
78,704 KB
testcase_21 AC 129 ms
78,764 KB
testcase_22 AC 127 ms
78,664 KB
testcase_23 AC 971 ms
142,388 KB
testcase_24 AC 864 ms
143,108 KB
testcase_25 AC 856 ms
143,608 KB
testcase_26 AC 881 ms
141,192 KB
testcase_27 AC 981 ms
142,796 KB
testcase_28 AC 595 ms
143,412 KB
testcase_29 AC 615 ms
143,588 KB
testcase_30 AC 735 ms
142,840 KB
testcase_31 AC 731 ms
143,556 KB
testcase_32 AC 750 ms
144,384 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

"""
C++ の std::map ぽいデータ構造
座標圧縮は前提
n: 取りうる最大値（最小値は 0）
self.dic を dict に取り換えることでメモリ節約できる
"""
class BIT: #0-indexed
    def __init__(self, n):
        self.size = n
        self.tree = [0]*(n+1)
        self.depth = n.bit_length()
        self.n0 = 1<<self.depth
#        self.element = [0]*(n+1)
    def get_sum(self, i): #a_0 + ... + a_{i} #閉区間
        s = 0; i += 1
        while i > 0:
            s += self.tree[i]
            i -= i & -i
        return s
    def query(self,l,r): #a_l + ... + a_r 閉区間
        return self.get_sum(r) - self.get_sum(l-1) 
    def add(self, i, x):
        i += 1
        while i <= self.size:
            self.tree[i] += x
            i += i & -i
        # self.element[i] += x
    #def get(self,i): return element[i]        
    def bisect_left(self,w):
        #和が w 以上になる最小の index
        #w が存在しない場合 self.size を返す
        if w <= 0: return 0
        x,k = 0,self.n0
        for _ in range(self.depth):
            k >>= 1
            if x+k <= self.size and self.tree[x+k] < w:
                w -= self.tree[x+k]
                x += k
        return x
    
class stdmap:
    def __init__(self, n):
        self.size = n+1
        self.keys = set()
        self.B = BIT(n+1) #存在すれば 1、しないなら 0
        self.dic = [0]*(n+1) # 値域
    
    def __contains__(self, k):
        return k in self.keys

    def insert(self,a,b): # 値 a に b を上書き
        if a not in self.keys:
            self.B.add(a,1)
            self.keys.add(a)
        self.dic[a] = b

    def remove(self,a): # a を取り除く
        self.keys.remove(a)
        self.B.add(a,-1)

    def lower_bound(self,k): # k 以上の最小のkeyを求める
        return self.B.bisect_left(self.B.get_sum(k-1)+1)
        
    def kth_key(self,k): # k 番目に小さい元のkeyを求める
        return self.B.bisect_left(k)

    def kth_value(self,k): # k 番目に小さい元のmap先を求める
        return self.dic[self.B.bisect_left(k)]

    def prev_key(self,k): #一個前の元のkeyを求める
        idx = self.B.get_sum(k)
        assert idx != 0
        return self.B.bisect_left(idx-1)

    def next_key(self,k):
        idx = self.B.get_sum(k)
        assert idx != self.size
        return self.B.bisect_left(idx+1)

    def __getitem__(self,item):
        return self.dic[item]


# coding: utf-8
# Your code here!
import sys
read = sys.stdin.read
readline = sys.stdin.readline

n,*a = map(int,read().split())

sa = sorted(a+[0]+[1<<29])
zaatu = {ai:i for i,ai in enumerate(sa)}


b1 = stdmap(n+2)
b2 = stdmap(n+2)
for i in range(n+2):
    b2.insert(i,1)

b1.insert(0,1)
b1.insert(n+1,1)


ans = INF = 1<<31
n += 1
for i,ai in enumerate(a):
    z = zaatu[ai]
    b2.remove(z)

    i1 = b1.lower_bound(z)
    i2 = b2.lower_bound(z)

    if i1 < n and i2 < n:
        r = sa[i1] + ai + sa[i2]
        #print(r,z,i1,i2,ai)
        ans = min(ans,r)
    
    i1 = b1.kth_key(2)
    i2 = b2.kth_key(2)

    if i1 < z and i2 < z:
        r = sa[i1] + ai + sa[i2]
        #print(b2.keys)
        #print(r,z,i1,i2,ai)
        ans = min(ans,r)

    
    b1.insert(z,1)


if ans==INF: print(-1)
else: print(ans)
0