結果

問題 No.2821 A[i] ← 2A[j] - A[i]
ユーザー ねしんねしん
提出日時 2024-07-26 23:40:30
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 4,351 bytes
コンパイル時間 347 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 201,108 KB
最終ジャッジ日時 2024-07-26 23:40:55
合計ジャッジ時間 24,890 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 44 ms
54,876 KB
testcase_01 TLE -
testcase_02 WA -
testcase_03 TLE -
testcase_04 TLE -
testcase_05 TLE -
testcase_06 TLE -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 TLE -
testcase_11 TLE -
testcase_12 TLE -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 TLE -
testcase_17 TLE -
testcase_18 TLE -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import time
def gcd(a,b):
  if b<=0:
    return max(a,0)
  else:
    return gcd(b,a%b)
#####segfunc#####
def segfunc(x, y):
    return gcd(max(x,y),min(x,y))
#################

#####ide_ele#####
ide_ele = 0
#################

class LazySegmentTree:
    """
    init(init_val, ide_ele): 配列init_valで初期化 O(N)
    add(l, r, x): 区間[l, r)にxを加算 O(logN)
    query(l, r): 区間[l, r)をsegfuncしたものを返す O(logN)
    """
    def __init__(self, init_val, segfunc, ide_ele):
        """
        init_val: 配列の初期値
        segfunc: 区間にしたい操作
        ide_ele: 単位元
        num: n以上の最小の2のべき乗
        data: 値配列(1-index)
        lazy: 遅延配列(1-index)
        """
        n = len(init_val)
        self.segfunc = segfunc
        self.ide_ele = ide_ele
        self.num = 1 << (n - 1).bit_length()
        self.data = [ide_ele] * 2 * self.num
        self.lazy = [0] * 2 * self.num
        # 配列の値を葉にセット
        for i in range(n):
            self.data[self.num + i] = init_val[i]
        # 構築していく
        for i in range(self.num - 1, 0, -1):
            self.data[i] = self.segfunc(self.data[2 * i], self.data[2 * i + 1])

    def gindex(self, l, r):
            """
            伝搬する対象の区間を求める
            lm: 伝搬する必要のある最大の左閉区間
            rm: 伝搬する必要のある最大の右開区間
            """
            l += self.num
            r += self.num
            lm = l >> (l & -l).bit_length()
            rm = r >> (r & -r).bit_length()

            while r > l:
                if l <= lm:
                    yield l
                if r <= rm:
                    yield r
                r >>= 1
                l >>= 1
            while l:
                yield l
                l >>= 1

    def propagates(self, *ids):
        """
        遅延伝搬処理
        ids: 伝搬する対象の区間 
        """
        for i in reversed(ids):
            v = self.lazy[i]
            if not v:
                continue
            self.lazy[2 * i] += v
            self.lazy[2 * i + 1] += v
            self.data[2 * i] += v
            self.data[2 * i + 1] += v
            self.lazy[i] = 0

    def add(self, l, r, x):
        """
        区間[l, r)の値にxを加算
        l, r: index(0-index)
        x: additional value
        """
        *ids, = self.gindex(l, r)
        l += self.num
        r += self.num
        while l < r:
            if l & 1:
                self.lazy[l] += x
                self.data[l] += x
                l += 1
            if r & 1:
                self.lazy[r - 1] += x
                self.data[r - 1] += x
            r >>= 1
            l >>= 1
        for i in ids:
            self.data[i] = self.segfunc(self.data[2 * i], self.data[2 * i + 1]) + self.lazy[i]


    def query(self, l, r):
        """
        [l, r)のsegfuncしたものを得る
        l: index(0-index)
        r: index(0-index)
        """
        *ids, = self.gindex(l, r)
        self.propagates(*ids)

        res = self.ide_ele

        l += self.num
        r += self.num
        while l < r:
            if l & 1:
                res = self.segfunc(res, self.data[l])
                l += 1
            if r & 1:
                res = self.segfunc(res, self.data[r - 1])
            l >>= 1
            r >>= 1
        return res
from collections import defaultdict
start=time.time()
N=int(input())
A=list(map(int,input().split()))
seg1 = LazySegmentTree([0]*(N+3), segfunc, ide_ele)
seg2 = LazySegmentTree([0]*(N+3), segfunc, ide_ele)
M=[]
ind=defaultdict(int)
for i in range(N):
  M.append((A[i],i))
M=sorted(M)
for i in range(N):
  ind[M[i][1]]=i+2
print(0)
if N==1:
  exit()
seg1.add(ind[0],ind[0]+1,A[0])
seg1.add(ind[1],ind[1]+1,A[1])
seg2.add(ind[0],ind[0]+1,-A[0])
seg2.add(ind[1],ind[1]+1,-A[1])
ans=abs(A[0]-A[1])
print(ans)
for i in range(2,N):
  if ans==1:
    for j in range(i,N):
    	print(1)
    exit()
  if time.time()-start>1.5:
  	exit()
  p=ind[i]
  #print(i,p)
  seg1.add(p,p+1,A[i])
  seg2.add(p,p+1,-A[i])
  seg1.add(p,N+2,-A[i])
  seg2.add(1,p,A[i])
  q=seg1.query(p,N+1)
  ans=gcd(max(ans,q),min(ans,q))
  r=seg2.query(1,p)
  ans=gcd(max(ans,r),min(ans,r))
  print(ans)
  seg1.add(p,N+2,A[i])
  seg2.add(1,p,-A[i])
0