結果
問題 | No.1391 ±1 Abs Sum |
ユーザー | kohei2019 |
提出日時 | 2021-02-12 23:00:58 |
言語 | PyPy3 (7.3.13) |
結果 |
AC
|
実行時間 | 1,284 ms / 2,000 ms |
コード長 | 3,429 bytes |
コンパイル時間 | 280 ms |
コンパイル使用メモリ | 87,216 KB |
実行使用メモリ | 137,440 KB |
最終ジャッジ日時 | 2023-09-27 06:11:35 |
合計ジャッジ時間 | 26,002 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 66 ms
71,272 KB |
testcase_01 | AC | 64 ms
71,172 KB |
testcase_02 | AC | 62 ms
71,092 KB |
testcase_03 | AC | 63 ms
71,356 KB |
testcase_04 | AC | 64 ms
71,400 KB |
testcase_05 | AC | 63 ms
71,488 KB |
testcase_06 | AC | 66 ms
71,400 KB |
testcase_07 | AC | 65 ms
71,676 KB |
testcase_08 | AC | 89 ms
76,872 KB |
testcase_09 | AC | 87 ms
76,740 KB |
testcase_10 | AC | 88 ms
76,528 KB |
testcase_11 | AC | 1,251 ms
136,672 KB |
testcase_12 | AC | 943 ms
116,044 KB |
testcase_13 | AC | 1,125 ms
129,104 KB |
testcase_14 | AC | 953 ms
104,732 KB |
testcase_15 | AC | 896 ms
108,396 KB |
testcase_16 | AC | 1,075 ms
117,988 KB |
testcase_17 | AC | 980 ms
114,256 KB |
testcase_18 | AC | 1,182 ms
129,140 KB |
testcase_19 | AC | 808 ms
111,876 KB |
testcase_20 | AC | 1,200 ms
134,144 KB |
testcase_21 | AC | 911 ms
116,868 KB |
testcase_22 | AC | 899 ms
118,692 KB |
testcase_23 | AC | 1,088 ms
121,492 KB |
testcase_24 | AC | 951 ms
120,288 KB |
testcase_25 | AC | 1,016 ms
121,360 KB |
testcase_26 | AC | 1,012 ms
120,488 KB |
testcase_27 | AC | 788 ms
111,340 KB |
testcase_28 | AC | 822 ms
111,300 KB |
testcase_29 | AC | 983 ms
117,788 KB |
testcase_30 | AC | 874 ms
120,296 KB |
testcase_31 | AC | 1,284 ms
137,440 KB |
testcase_32 | AC | 95 ms
99,720 KB |
testcase_33 | AC | 98 ms
98,268 KB |
testcase_34 | AC | 790 ms
130,568 KB |
testcase_35 | AC | 64 ms
71,492 KB |
testcase_36 | AC | 868 ms
136,560 KB |
ソースコード
import sys import math sys.setrecursionlimit(10**7) class lazysegki(): #区間和のセグメント木、葉の数N,要素ls def __init__(self,N,ls,mode='sum'): self.default = 0 self.func = (lambda x, y: x + y) self.N = N self.K = (N-1).bit_length() self.N0 = 1 << self.K self.dat = [self.default]*(2**(self.K+1)) #全体の配列を用意(maxなら-inf,minなら+infで未構築を表現) self.lazy = [0]*(2**(self.K+1)) #遅延評価 for i in range(self.N): #葉の構築 self.dat[2**self.K+i] = ls[i] self.build() def build(self): for j in range(self.N0-1,0,-1): self.dat[j] = self.func(self.dat[j<<1],self.dat[j<<1|1]) #親が持つ条件 def leafvalue(self,x): #x番目の値を出力 return self.query(x,x+1) def update_add(self,x,y): #x番目にyを足す return self.updatel_add(x,x+1,y) def gindex(self,l, r): #伝播するインデックス列挙 L = l + self.N0; R = r + self.N0 lm = (L // (L & -L)) >> 1 rm = (R // (R & -R)) >> 1 lsindex = [] while L < R: if R <= rm: lsindex.append(R) if L <= lm: lsindex.append(L) L >>= 1; R >>= 1 while L: lsindex.append(L) L >>= 1 return lsindex def propagates(self,ids): for i in reversed(ids): v = self.lazy[i] if not v: continue self.lazy[i<<1] += v//2; self.lazy[i<<1|1] += v//2 self.dat[i<<1] += v//2; self.dat[i<<1|1] += v//2 self.lazy[i] = 0 def updatel_add(self,l, r, x):#区間にyを足す self.propagates(self.gindex(l, r)) L = self.N0 + l R = self.N0 + r ii = 0 while L < R: if L & 1: self.lazy[L] += x*(2**ii) self.dat[L] += x*(2**ii) L += 1 if R & 1: R -= 1 self.lazy[R] += x*(2**ii) self.dat[R] += x*(2**ii) L >>= 1 R >>= 1 ii += 1 for i in self.gindex(l, r): self.dat[i] = self.func(self.dat[i<<1],self.dat[i<<1|1])+self.lazy[i] def query(self,l, r): self.propagates(self.gindex(l, r)) L = self.N0 + l R = self.N0 + r vL = self.default vR = self.default while L < R: if L & 1: vL = self.func(vL,self.dat[L]) L += 1 if R & 1: R -= 1 vR = self.func(self.dat[R],vR) L >>= 1 R >>= 1 return self.func(vL,vR) N,K = map(int,input().split()) lsA = list(map(int,input().split())) minA = min(lsA) lsA2 = [i-minA for i in lsA] lsv = [0]*(N) lsv[0] = sum(lsA2) lsc = [0] c = 0 for i in range(1,N): lsv[i] = lsv[i-1]+(lsA2[i]-c)*(2*i-N) lsc.append(lsA2[i]-c) c += lsA2[i]-c if K == 0: print(-max(lsv)) exit() SG = lazysegki(N,lsA2) l = 0 r = K lsq = [sum(lsA2[:K])] for i in range(1,N): SG.updatel_add(0,i,lsc[i]) SG.updatel_add(i,N,-lsc[i]) while r < N and SG.leafvalue(l) > SG.leafvalue(r): l += 1 r += 1 lsq.append(SG.query(l,r)) lsans = [] for i in range(N): m = -(lsv[i]-lsq[i]) m += lsq[i] lsans.append(m) print(min(lsans))