結果

問題 No.1219 Mancala Combo
ユーザー titia
提出日時 2020-09-04 21:44:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 151 ms / 2,000 ms
コード長 823 bytes
コンパイル時間 952 ms
コンパイル使用メモリ 82,020 KB
実行使用メモリ 109,968 KB
最終ジャッジ日時 2024-11-26 12:24:50
合計ジャッジ時間 3,138 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

N=int(input())
A=list(map(int,input().split()))

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

for i in range(N):
    SEG[i+seg_el+1]=A[i]

def getvalue(n,seg_el): # 一点の値を取得
    i=n+seg_el

    ANS=SEG[i]
    i>>=1 # 子ノードへ
    
    while i!=0:
        ANS+=SEG[i]
        i>>=1

    return ANS

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

    while L<R:
        if L & 1:
            SEG[L]+=x
            L+=1

        if R & 1:
            R-=1
            SEG[R]+=x
        L>>=1
        R>>=1

for i in range(N,0,-1):
    V=getvalue(i,seg_el)
    #print(V,i)
    if V%i!=0:
        print("No")
        break
    else:
        updates(0,i,V//i)
else:
    print("Yes")
0