結果
問題 | No.2821 A[i] ← 2A[j] - A[i] |
ユーザー | ねしん |
提出日時 | 2024-07-26 23:30:54 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 4,220 bytes |
コンパイル時間 | 281 ms |
コンパイル使用メモリ | 82,148 KB |
実行使用メモリ | 236,012 KB |
最終ジャッジ日時 | 2024-07-26 23:30:58 |
合計ジャッジ時間 | 4,027 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 39 ms
61,512 KB |
testcase_01 | TLE | - |
testcase_02 | -- | - |
testcase_03 | -- | - |
testcase_04 | -- | - |
testcase_05 | -- | - |
testcase_06 | -- | - |
testcase_07 | -- | - |
testcase_08 | -- | - |
testcase_09 | -- | - |
testcase_10 | -- | - |
testcase_11 | -- | - |
testcase_12 | -- | - |
testcase_13 | -- | - |
testcase_14 | -- | - |
testcase_15 | -- | - |
testcase_16 | -- | - |
testcase_17 | -- | - |
testcase_18 | -- | - |
testcase_19 | -- | - |
testcase_20 | -- | - |
testcase_21 | -- | - |
testcase_22 | -- | - |
testcase_23 | -- | - |
testcase_24 | -- | - |
testcase_25 | -- | - |
testcase_26 | -- | - |
testcase_27 | -- | - |
testcase_28 | -- | - |
testcase_29 | -- | - |
testcase_30 | -- | - |
testcase_31 | -- | - |
testcase_32 | -- | - |
testcase_33 | -- | - |
ソースコード
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 N=int(input()) A=list(map(int,input().split())) seg1 = LazySegmentTree([0]*(2*N), segfunc, ide_ele) seg2 = LazySegmentTree([0]*(2*N), 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): 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])