結果

問題 No.3094 Stapler
コンテスト
ユーザー timi
提出日時 2025-04-08 02:10:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,610 ms / 2,000 ms
コード長 6,095 bytes
コンパイル時間 237 ms
コンパイル使用メモリ 82,564 KB
実行使用メモリ 287,112 KB
最終ジャッジ日時 2025-10-23 22:28:11
合計ジャッジ時間 76,147 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 72
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
#N,Q=map(int,input().split())
N=int(input())
mod=998244353
class lazy_segtree():
    def update(self,k):self.d[k]=self.op(self.d[2*k],self.d[2*k+1])
    def all_apply(self,k,f):
        self.d[k]=self.mapping(f,self.d[k])
        if (k<self.size):self.lz[k]=self.composition(f,self.lz[k])
    def push(self,k):
        self.all_apply(2*k,self.lz[k])
        self.all_apply(2*k+1,self.lz[k])
        self.lz[k]=self.identity
    def __init__(self,V,OP,E,MAPPING,COMPOSITION,ID):
        self.n=len(V)
        self.log=(self.n-1).bit_length()
        self.size=1<<self.log
        self.d=[E for i in range(2*self.size)]
        self.lz=[ID for i in range(self.size)]
        self.e=E
        self.op=OP
        self.mapping=MAPPING
        self.composition=COMPOSITION
        self.identity=ID
        for i in range(self.n):self.d[self.size+i]=V[i]
        for i in range(self.size-1,0,-1):self.update(i)
    def set(self,p,x):
        assert 0<=p and p<self.n
        p+=self.size
        for i in range(self.log,0,-1):self.push(p>>i)
        self.d[p]=x
        for i in range(1,self.log+1):self.update(p>>i)
    def get(self,p):
        assert 0<=p and p<self.n
        p+=self.size
        for i in range(self.log,0,-1):self.push(p>>i)
        return self.d[p]
    def prod(self,l,r):
        assert 0<=l and l<=r and r<=self.n
        if l==r:return self.e
        l+=self.size
        r+=self.size
        for i in range(self.log,0,-1):
            if (((l>>i)<<i)!=l):self.push(l>>i)
            if (((r>>i)<<i)!=r):self.push(r>>i)
        sml,smr=self.e,self.e
        while(l<r):
            if l&1:
                sml=self.op(sml,self.d[l])
                l+=1
            if r&1:
                r-=1
                smr=self.op(self.d[r],smr)
            l>>=1
            r>>=1
        return self.op(sml,smr)
    def all_prod(self):return self.d[1]
    def apply_point(self,p,f):
        assert 0<=p and p<self.n
        p+=self.size
        for i in range(self.log,0,-1):self.push(p>>i)
        self.d[p]=self.mapping(f,self.d[p])
        for i in range(1,self.log+1):self.update(p>>i)
    def apply(self,l,r,f):
        assert 0<=l and l<=r and r<=self.n
        if l==r:return
        l+=self.size
        r+=self.size
        for i in range(self.log,0,-1):
            if (((l>>i)<<i)!=l):self.push(l>>i)
            if (((r>>i)<<i)!=r):self.push((r-1)>>i)
        l2,r2=l,r
        while(l<r):
            if (l&1):
                self.all_apply(l,f)
                l+=1
            if (r&1):
                r-=1
                self.all_apply(r,f)
            l>>=1
            r>>=1
        l,r=l2,r2
        for i in range(1,self.log+1):
            if (((l>>i)<<i)!=l):self.update(l>>i)
            if (((r>>i)<<i)!=r):self.update((r-1)>>i)
    def max_right(self,l,g):
        assert 0<=l and l<=self.n
        assert g(self.e)
        if l==self.n:return self.n
        l+=self.size
        for i in range(self.log,0,-1):self.push(l>>i)
        sm=self.e
        while(1):
            while(l%2==0):l>>=1
            if not(g(self.op(sm,self.d[l]))):
                while(l<self.size):
                    self.push(l)
                    l=(2*l)
                    if (g(self.op(sm,self.d[l]))):
                        sm=self.op(sm,self.d[l])
                        l+=1
                return l-self.size
            sm=self.op(sm,self.d[l])
            l+=1
            if (l&-l)==l:break
        return self.n
    def min_left(self,r,g):
        assert (0<=r and r<=self.n)
        assert g(self.e)
        if r==0:return 0
        r+=self.size
        for i in range(self.log,0,-1):self.push((r-1)>>i)
        sm=self.e
        while(1):
            r-=1
            while(r>1 and (r%2)):r>>=1
            if not(g(self.op(self.d[r],sm))):
                while(r<self.size):
                    self.push(r)
                    r=(2*r+1)
                    if g(self.op(self.d[r],sm)):
                        sm=self.op(self.d[r],sm)
                        r-=1
                return r+1-self.size
            sm=self.op(self.d[r],sm)
            if (r&-r)==r:break
        return 0
    
# 更新 と 取得
# 最初の V -> 更新 f を行った後の 状態を mapping に定義 -> operate によって取得

# 今回は、値, 区間の長さを持たせたらうまく行きそう
# f(l, r) と r - l


# op : S[l, m) と S[m, r) を引数として S[l, r) を求める関数
# e  : 区間長 0 の区間に対応する情報
# mapping : 更新クエリ f を A[l, r) に対応させた後の S[l, r) を表す
# composition : 更新クエリ g, f をこの順で同じ区間に適応させることを 1 つの更新クエリとした際のものを表す
# id : 恒等写像を表す更新クエリ、すなわち何もしないクエリ 更新する値として取りえない値を取れば良い

def operate(a,b):
  #最小値と最小値の個数
  al,ar=a 
  bl,br=b
  if ar+br==0:
    return (min(al,bl),0) 
  elif ar==0:
    return b 
  elif br==0:
    return a
    
  c=min(al,bl)
  cc=0
  if c==al:
    cc+=ar
  if c==bl:
    cc+=br
  #print(a,b,(c,cc))
  return (c,cc)

# 更新クエリ f を各ノードが持つ data の値 x に対して作用させる関数
def mapping(f,x):
    if f == id:
      return x
    else:
      l,r=x
      if r>0:
        return (l+f,r)
      else:
        return x

# 更新クエリ g,f をこの順で同じ区間に適応させることを 1 つの更新クエリとした際のものを表すもの
def composition(f,g):
    if f == id:
        return g
    elif g==id:
        return f
    else:
        return f+g

e = (0,0)
id = -10**10

# V,OP,E,MAPPING,COMPOSITION,ID
L=[]
for i in range(N):
  L.append([0,1])
seg=lazy_segtree(L,operate,e,mapping,composition,id)
Q=int(input())
A=[]

for _ in range(Q):
  #l,r,d=map(int, input().split())
  qs=list(map(int, input().split()))
  A.append(qs)
  if qs[0]==1:
    _,l,r=qs
    seg.apply(l,r,1)
  elif qs[0]==2:
    _,q=qs
    _,l,r=A[q-1]
    seg.apply(l,r,-1)
  else:
    c=seg.all_prod()[1]
    print(c)
0