結果
問題 | 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 |
ソースコード
class SegTree:def __init__(self, N, id, f):self.n = 2**(N-1).bit_length()self.unitX = idself.X = [id] * (self.n * 2)self.f = fself.N = Ndef update(self, i, x):i += self.nself.X[i] = xi >>= 1while i:self.X[i] = self.f(self.X[i*2], self.X[i*2|1])i >>= 1def getvalue(self, i):return self.X[i + self.n]def query(self, l, r):l += self.nr += self.nal = self.unitXar = self.unitXwhile l < r:if l & 1:al = self.f(al, self.X[l])l += 1if r & 1:r -= 1ar = self.f(self.X[r], ar)l >>= 1r >>= 1return self.f(al, ar)def max_right(self, l, g):l += self.nll = l // (l & -l)ans = self.unitXwhile g(self.f(ans, self.X[ll])):ans = self.f(ans, self.X[ll])ll += 1while ~ll & 1:ll >>= 1if ll == 1:return self.Nwhile ll < self.n:ll <<= 1if g(self.f(ans, self.X[ll])):ans = self.f(ans, self.X[ll])ll += 1return ll-self.ndef min_left(self, r, g):r += self.nrr = max(r // (~r & -~r), 1)ans = self.unitXwhile g(self.f(self.X[rr], ans)):ans = self.f(self.X[rr], ans)rr -= 1while rr & 1:rr >>= 1if rr == 0:return -1while rr < self.n:rr <<= 1if g(self.f(self.X[rr+1], ans)):ans = self.f(self.X[rr+1], ans)else:rr += 1return rr - self.ndef f(a,b):am1,am2,aM=abm1,bm2,bM=bif am1<bm1:m1=am1m2=min(am2,bm1)else:m1=bm1m2=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=0for i in range(n-1):j=seg.max_right(i,lambda x:x[2]<=x[0]+x[1])ans+=j-i-1print(ans)