結果

問題 No.1435 Mmm......
ユーザー tyawanmusityawanmusi
提出日時 2020-10-22 16:56:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,515 ms / 2,000 ms
コード長 1,908 bytes
コンパイル時間 323 ms
コンパイル使用メモリ 87,048 KB
実行使用メモリ 139,656 KB
最終ジャッジ日時 2023-09-28 14:41:12
合計ジャッジ時間 27,659 ms
ジャッジサーバーID
(参考情報)
judge14 / judge15
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 77 ms
71,544 KB
testcase_01 AC 76 ms
71,484 KB
testcase_02 AC 76 ms
71,324 KB
testcase_03 AC 74 ms
71,296 KB
testcase_04 AC 75 ms
71,220 KB
testcase_05 AC 75 ms
71,328 KB
testcase_06 AC 1,117 ms
124,836 KB
testcase_07 AC 1,231 ms
127,180 KB
testcase_08 AC 1,293 ms
130,640 KB
testcase_09 AC 1,228 ms
127,440 KB
testcase_10 AC 1,156 ms
125,508 KB
testcase_11 AC 1,123 ms
126,224 KB
testcase_12 AC 1,064 ms
124,312 KB
testcase_13 AC 1,137 ms
125,196 KB
testcase_14 AC 913 ms
106,712 KB
testcase_15 AC 1,287 ms
130,348 KB
testcase_16 AC 1,246 ms
127,852 KB
testcase_17 AC 980 ms
109,216 KB
testcase_18 AC 1,292 ms
130,140 KB
testcase_19 AC 1,097 ms
125,148 KB
testcase_20 AC 1,271 ms
129,280 KB
testcase_21 AC 1,515 ms
136,532 KB
testcase_22 AC 451 ms
116,436 KB
testcase_23 AC 1,113 ms
139,068 KB
testcase_24 AC 1,119 ms
139,152 KB
testcase_25 AC 1,133 ms
138,960 KB
testcase_26 AC 1,146 ms
139,656 KB
testcase_27 AC 1,160 ms
139,132 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

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

  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)

  def max_right(self, l, g):
    l += self.num
    ll = l // (l & -l)
    ans = self.unit
    while g(self.f(ans, self.seg[ll])):
      ans = self.f(ans, self.seg[ll])
      ll += 1
      while ~ll & 1:
        ll >>= 1
      if ll == 1:
        return self.n
    while ll < self.num:
      ll <<= 1
      if g(self.f(ans, self.seg[ll])):
        ans = self.f(ans, self.seg[ll])
        ll += 1
    return ll-self.num

MAX=10**9+1
MIN=0

hassh=lambda x,y,z: x*MAX*MAX+y*MAX+z

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

def g(x):
  xm1,xm2xM=divmod(x,MAX*MAX)
  xm2,xM=divmod(xm2xM,MAX)
  return xM<=xm1+xm2

n=int(input())
p=list(map(int,input().split()))
seg=SegmentTree(n,[hassh(i,10**9,i)for i in p],hassh(10**9,10**9,0),f)
ans=0
for l in range(n-1):
  r=seg.max_right(l, g)
  ans+=r-l-1
print(ans)
0