結果

問題 No.1673 Lamps on a line
ユーザー titiatitia
提出日時 2021-09-10 21:43:16
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 2,470 bytes
コンパイル時間 304 ms
コンパイル使用メモリ 82,040 KB
実行使用メモリ 97,788 KB
最終ジャッジ日時 2024-06-11 23:07:40
合計ジャッジ時間 3,557 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
54,212 KB
testcase_01 AC 39 ms
54,304 KB
testcase_02 WA -
testcase_03 AC 157 ms
78,608 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 74 ms
87,252 KB
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 AC 332 ms
97,788 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

N,Q=map(int,input().split())

# 非再帰遅延伝搬セグ木

seg_el=1<<(N.bit_length()) # Segment treeの台の要素数
seg_height=1+N.bit_length() # Segment treeの高さ
SEG=[0]*(2*seg_el) # 1-indexedなので、要素数2*seg_el.Segment treeの初期値で初期化
LAZY=[0]*(2*seg_el) # 1-indexedなので、要素数2*seg_el.Segment treeの初期値で初期化

def indexes(L,R): # 遅延伝搬すべきノードのリストを返す. （つまり, updateやgetvaluesで見るノードより上にあるノードたち）
    INDLIST=[]

    R-=1
    
    L>>=1
    R>>=1

    while L!=R:
        if L>R:
            INDLIST.append(L)
            L>>=1
        else:
            INDLIST.append(R)
            R>>=1

    while L!=0:
        INDLIST.append(L)
        L>>=1

    return INDLIST

def updates(l,r):
        
    L=l+seg_el
    R=r+seg_el

    L//=(L & (-L))
    R//=(R & (-R))

    UPIND=indexes(L,R)
    
    for ind in UPIND[::-1]: # 遅延伝搬
        if LAZY[ind]==1:
            LAZY[ind<<1]=LAZY[1+(ind<<1)]=1
            SEG[ind<<1]=(1<<(seg_height - ((ind<<1).bit_length()))) - SEG[ind<<1]
            SEG[1+(ind<<1)]=(1<<(seg_height - ((1+(ind<<1)).bit_length()))) - SEG[1+(ind<<1)]
            LAZY[ind]=0

    while L!=R:
        if L > R:
            SEG[L]=(1<<(seg_height - (L.bit_length()))) - SEG[L]
            LAZY[L]=1
            L+=1
            L//=(L & (-L))

        else:
            R-=1
            SEG[R]=(1<<(seg_height - (R.bit_length()))) - SEG[R]
            LAZY[R]=1
            R//=(R & (-R))

    for ind in UPIND:
        SEG[ind]=SEG[ind<<1]+SEG[1+(ind<<1)]

def getvalues(l,r): # 区間[l,r)に関する和を調べる

    L=l+seg_el
    R=r+seg_el

    L//=(L & (-L))
    R//=(R & (-R))

    UPIND=indexes(L,R)
    
    for ind in UPIND[::-1]: # 遅延伝搬
        if LAZY[ind]==1:
            LAZY[ind<<1]=LAZY[1+(ind<<1)]=1
            SEG[ind<<1]=(1<<(seg_height - ((ind<<1).bit_length()))) - SEG[ind<<1]
            SEG[1+(ind<<1)]=(1<<(seg_height - ((1+(ind<<1)).bit_length()))) - SEG[1+(ind<<1)]
            LAZY[ind]=0
            
    ANS=0

    while L!=R:
        if L > R:
            ANS+=SEG[L]
            L+=1
            L//=(L & (-L))

        else:
            R-=1
            ANS+=SEG[R]
            R//=(R & (-R))

    return ANS

for queries in range(Q):
    l,r=map(int,input().split())
    updates(l,r+1)

    print(getvalues(1,N+1))
0