結果

問題 No.1435 Mmm......
ユーザー あかりき
提出日時 2021-03-20 18:20:16
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 501 ms / 2,000 ms
コード長 1,915 bytes
コンパイル時間 156 ms
コンパイル使用メモリ 82,584 KB
実行使用メモリ 122,484 KB
最終ジャッジ日時 2024-11-21 07:02:22
合計ジャッジ時間 11,326 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

class SegTree:
def __init__(self, N, id, f):
self.n = 2**(N-1).bit_length()
self.unitX = id
self.X = [id] * (self.n * 2)
self.f = f
self.N = N
def update(self, i, x):
i += self.n
self.X[i] = x
i >>= 1
while i:
self.X[i] = self.f(self.X[i*2], self.X[i*2|1])
i >>= 1
def getvalue(self, i):
return self.X[i + self.n]
def query(self, l, r):
l += self.n
r += self.n
al = self.unitX
ar = self.unitX
while l < r:
if l & 1:
al = self.f(al, self.X[l])
l += 1
if r & 1:
r -= 1
ar = self.f(self.X[r], ar)
l >>= 1
r >>= 1
return self.f(al, ar)
def max_right(self, l, g):
l += self.n
ll = l // (l & -l)
ans = self.unitX
while g(self.f(ans, self.X[ll])):
ans = self.f(ans, self.X[ll])
ll += 1
while ~ll & 1:
ll >>= 1
if ll == 1:
return self.N
while ll < self.n:
ll <<= 1
if g(self.f(ans, self.X[ll])):
ans = self.f(ans, self.X[ll])
ll += 1
return ll-self.n
def min_left(self, r, g):
r += self.n
rr = max(r // (~r & -~r), 1)
ans = self.unitX
while g(self.f(self.X[rr], ans)):
ans = self.f(self.X[rr], ans)
rr -= 1
while rr & 1:
rr >>= 1
if rr == 0:
return -1
while rr < self.n:
rr <<= 1
if g(self.f(self.X[rr+1], ans)):
ans = self.f(self.X[rr+1], ans)
else:
rr += 1
return rr - self.n
def f(a,b):
am1,am2,aM=a
bm1,bm2,bM=b
if am1<bm1:
m1=am1
m2=min(am2,bm1)
else:
m1=bm1
m2=min(bm2,am1)
M=max(aM,bM)
return (m1,m2,M)
n=int(input())
a=list(map(int,input().split()))
seg=SegTree(n,(10**18,10**18,-10**18),f)
for i in range(n):
seg.update(i,(a[i],10**18,a[i]))
ans=0
for i in range(n-1):
j=seg.max_right(i,lambda x:x[2]<=x[0]+x[1])
ans+=j-i-1
print(ans)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0