結果
問題 | No.1095 Smallest Kadomatsu Subsequence |
ユーザー |
![]() |
提出日時 | 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 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
"""C++ の std::map ぽいデータ構造座標圧縮は前提n: 取りうる最大値(最小値は 0)self.dic を dict に取り換えることでメモリ節約できる"""class BIT: #0-indexeddef __init__(self, n):self.size = nself.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 += 1while i > 0:s += self.tree[i]i -= i & -ireturn sdef query(self,l,r): #a_l + ... + a_r 閉区間return self.get_sum(r) - self.get_sum(l-1)def add(self, i, x):i += 1while i <= self.size:self.tree[i] += xi += 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 0x,k = 0,self.n0for _ in range(self.depth):k >>= 1if x+k <= self.size and self.tree[x+k] < w:w -= self.tree[x+k]x += kreturn xclass stdmap:def __init__(self, n):self.size = n+1self.keys = set()self.B = BIT(n+1) #存在すれば 1、しないなら 0self.dic = [0]*(n+1) # 値域def __contains__(self, k):return k in self.keysdef insert(self,a,b): # 値 a に b を上書きif a not in self.keys:self.B.add(a,1)self.keys.add(a)self.dic[a] = bdef 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 != 0return self.B.bisect_left(idx-1)def next_key(self,k):idx = self.B.get_sum(k)assert idx != self.sizereturn self.B.bisect_left(idx+1)def __getitem__(self,item):return self.dic[item]# coding: utf-8# Your code here!import sysread = sys.stdin.readreadline = sys.stdin.readlinen,*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<<31n += 1for 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)