結果

問題 No.230 Splarraay ｽﾌﾟﾗﾚｪｰｲ
ユーザー timi
提出日時 2020-12-16 14:07:53
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 4,630 ms / 5,000 ms
コード長 1,943 bytes
コンパイル時間 172 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 39,172 KB
最終ジャッジ日時 2024-09-20 05:03:43
合計ジャッジ時間 29,450 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

N=int(input())
N
Q=int(input())
A,B=[],[]
D=[list(map(int, input().split())) for i in range(Q)]
# N: 処理する区間の長さ
LV = (N-1).bit_length()
N0 = 2**LV
data = [0]*(2*N0)
lazy = [None]*(2*N0)

def gindex(l, r):
    L = (l + N0) >> 1; R = (r + N0) >> 1
    lc = 0 if l & 1 else (L & -L).bit_length()
    rc = 0 if r & 1 else (R & -R).bit_length()
    for i in range(LV):
        if rc <= i:
            yield R
        if L < R and lc <= i:
            yield L
        L >>= 1; R >>= 1

# 遅延伝搬処理
def propagates(*ids):
    for i in reversed(ids):
        v = lazy[i-1]
        if v is None:
            continue
        lazy[2*i-1] = lazy[2*i] = data[2*i-1] = data[2*i] = v >> 1
        lazy[i-1] = None

# 区間[l, r)をxに更新
def update(l, r, x):
    *ids, = gindex(l, r)
    propagates(*ids)

    L = N0 + l; R = N0 + r
    v = x
    while L < R:
        if R & 1:
            R -= 1
            lazy[R-1] = data[R-1] = v
        if L & 1:
            lazy[L-1] = data[L-1] = v
            L += 1
        L >>= 1; R >>= 1; v <<= 1
    for i in ids:
        data[i-1] = data[2*i-1] + data[2*i]

# 区間[l, r)内の合計を求める
def query(l, r):
    propagates(*gindex(l, r))
    L = N0 + l; R = N0 + r
    s = 0
    while L < R:
        if R & 1:
            R -= 1
            s += data[R-1]
        if L & 1:
            s += data[L-1]
            L += 1
        L >>= 1; R >>= 1
    return s

for i in range(Q):
  x,l,r=D[i]
  if x==1:
    update(l,r+1,1)
  elif x==2:
    update(l,r+1,0)
  else:
    d=query(l,r+1)
    A.append(d)
d=query(0,N)
A.append(d)

LV = (N-1).bit_length()
N0 = 2**LV
data = [0]*(2*N0)
lazy = [None]*(2*N0)
for i in range(Q):
  x,l,r=D[i]
  if x==2:
    update(l,r+1,1)
  elif x==1:
    update(l,r+1,0)
  else:
    d=query(l,r+1)
    B.append(d)
d=query(0,N)
B.append(d)
a,b=A[-1],B[-1]
for i in range(len(A)-1):
  if A[i]>B[i]:
    a+=A[i]
  elif A[i]<B[i]:
    b+=B[i]
print(a,b)
0