結果

問題 No.1435 Mmm......
ユーザー tyawanmusityawanmusi
提出日時 2020-10-20 18:50:56
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,494 bytes
コンパイル時間 203 ms
コンパイル使用メモリ 82,608 KB
実行使用メモリ 107,260 KB
最終ジャッジ日時 2024-07-21 09:20:05
合計ジャッジ時間 4,068 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 TLE -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

class SegmentTree:
  def __init__(self, n, p, unit, f):
    self.num = 2**((n-1).bit_length())
    self.seg = [unit]*(2*self.num)
    for i in range(n):
      self.seg[i+self.num] = p[i]
    for i in range(self.num-1, 0, -1):
      self.seg[i] = f(self.seg[i << 1], self.seg[(i << 1)+1])
    self.f = f
    self.unit = unit

  def update(self, i, x):
    i += self.num
    self.seg[i] = x
    while i:
      i >>= 1
      self.seg[i] = self.f(self.seg[i << 1], self.seg[(i << 1)+1])

  def query(self, l, r):
    ansl = ansr = self.unit
    l += self.num
    r += self.num-1
    if l == r:
      return self.seg[l]
    while l < r:
      if l & 1:
        ansl = self.f(ansl, self.seg[l])
        l += 1
      if ~r & 1:
        ansr = self.f(self.seg[r], ansr)
        r -= 1
      l >>= 1
      r >>= 1
    if l == r:
      ansl = self.f(ansl, self.seg[l])
    return self.f(ansl, ansr)

# N<=3000を前提とする
hassh=lambda x,y,z: x*3001**2+y*3001+z

def f(x,y):
  xm1,xm2xM=divmod(x,3001*3001)
  xm2,xM=divmod(xm2xM,3001)
  ym1,ym2yM=divmod(y,3001*3001)
  ym2,yM=divmod(ym2yM,3001)
  M=max(xM,yM)
  if xm1<ym1:
    m1=xm1
    m2=min(ym1,xm2)
  else:
    m1=ym1
    m2=min(xm1,ym2)
  return hassh(m1,m2,M)

n=int(input())
p=list(map(int,input().split()))
seg=SegmentTree(n,[hassh(i,3000,i)for i in p],hassh(3000,3000,0),f)
ans=0
for l in range(n-2):
  for r in range(l+2,n):
    m1,m2M=divmod(seg.query(l,r+1),3001*3001)
    m2,M=divmod(m2M,3001)
    if M<=m1+m2:
      ans+=1
print(ans)
0