結果
| 問題 |
No.1375 Divide and Update
|
| コンテスト | |
| ユーザー |
👑 Kazun
|
| 提出日時 | 2021-02-05 22:42:50 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 321 ms / 2,000 ms |
| コード長 | 2,020 bytes |
| コンパイル時間 | 158 ms |
| コンパイル使用メモリ | 81,980 KB |
| 実行使用メモリ | 186,020 KB |
| 最終ジャッジ日時 | 2024-07-02 13:24:51 |
| 合計ジャッジ時間 | 5,880 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 19 |
ソースコード
import heapq
"""参考元
https://tsubo.hatenablog.jp/entry/2020/06/15/124657
"""
class Heap_Dict:
def __init__(self,Mode=True):
"""
Mode:True→最小値,False→最大値
"""
self.heap=[]
self.dict={}
self.Mode=Mode
def __bool__(self):
return bool(self.heap)
def insert(self,x):
if self.Mode and not self.is_exist(x):
heapq.heappush(self.heap,x)
elif not self.Mode and not self.is_exist(x):
heapq.heappush(self.heap,-x)
if x in self.dict:
self.dict[x]+=1
else:
self.dict[x]=1
def erase(self,x):
assert (x in self.dict) and (self.dict[x])
self.dict[x]-=1
while self.heap:
y=self.heap[0]
if not self.Mode:y=-y
if self.dict[y]==0:
heapq.heappop(self.heap)
else:
break
def is_exist(self,x):
return (x in self.dict) and (self.dict[x])
def get_min(self):
assert self.Mode
if self.heap:
return self.heap[0]
else:
return float("inf")
def get_max(self):
assert not self.Mode
if self.heap:
return -self.heap[0]
else:
return -float("inf")
def count(self,x):
if x not in self.dict:
return 0
else:
return self.dict[x]
#================================================
def gamma(P):
U=Heap_Dict()
U.insert(0);U.insert(P[0])
G=[-inf]*N
G[0]=P[0]
alpha=P[0]
for i in range(1,N):
alpha+=P[i]
G[i]=max(G[i-1],alpha-U.get_min())
U.insert(alpha)
return G
#================================================
import sys
input=sys.stdin.readline
N,X,Y=map(int,input().split())
A=list(map(int,input().split()))
A_sum=sum(A)
inf=float("inf")
P=gamma([X-a for a in A])
Q=gamma([Y-a for a in A][::-1])[::-1]
for i in range(1,N-1):
print(A_sum+P[i-1]+Q[i+1])
Kazun