結果
問題 | No.1391 ±1 Abs Sum |
ユーザー | persimmon-persimmon |
提出日時 | 2021-02-13 11:16:01 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 154 ms / 2,000 ms |
コード長 | 943 bytes |
コンパイル時間 | 276 ms |
コンパイル使用メモリ | 86,896 KB |
実行使用メモリ | 104,156 KB |
最終ジャッジ日時 | 2023-09-27 19:33:54 |
合計ジャッジ時間 | 7,895 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 71 ms
71,224 KB |
testcase_01 | AC | 72 ms
71,404 KB |
testcase_02 | AC | 72 ms
71,144 KB |
testcase_03 | AC | 73 ms
71,292 KB |
testcase_04 | AC | 72 ms
71,264 KB |
testcase_05 | AC | 71 ms
71,184 KB |
testcase_06 | AC | 72 ms
71,216 KB |
testcase_07 | AC | 71 ms
71,416 KB |
testcase_08 | AC | 73 ms
71,080 KB |
testcase_09 | AC | 71 ms
71,436 KB |
testcase_10 | AC | 73 ms
71,344 KB |
testcase_11 | AC | 154 ms
99,672 KB |
testcase_12 | AC | 134 ms
101,396 KB |
testcase_13 | AC | 146 ms
98,292 KB |
testcase_14 | AC | 123 ms
93,416 KB |
testcase_15 | AC | 124 ms
95,640 KB |
testcase_16 | AC | 136 ms
101,536 KB |
testcase_17 | AC | 128 ms
98,524 KB |
testcase_18 | AC | 143 ms
99,844 KB |
testcase_19 | AC | 128 ms
98,372 KB |
testcase_20 | AC | 153 ms
100,340 KB |
testcase_21 | AC | 136 ms
100,540 KB |
testcase_22 | AC | 132 ms
100,744 KB |
testcase_23 | AC | 139 ms
104,156 KB |
testcase_24 | AC | 131 ms
103,756 KB |
testcase_25 | AC | 137 ms
103,888 KB |
testcase_26 | AC | 131 ms
103,896 KB |
testcase_27 | AC | 118 ms
95,008 KB |
testcase_28 | AC | 122 ms
94,884 KB |
testcase_29 | AC | 134 ms
103,804 KB |
testcase_30 | AC | 128 ms
97,620 KB |
testcase_31 | AC | 148 ms
99,988 KB |
testcase_32 | AC | 95 ms
90,980 KB |
testcase_33 | AC | 98 ms
95,204 KB |
testcase_34 | AC | 131 ms
99,820 KB |
testcase_35 | AC | 71 ms
71,256 KB |
testcase_36 | AC | 130 ms
101,312 KB |
ソースコード
def main1(n,k,a): if k<=1: x0=a[0] x1=a[-1] ans0=0 ans1=0 for x in a: ans0-=abs(x-x0) ans1-=abs(x-x1) return min(ans0,ans1) l,r=0,k # [l,r):+1にする箇所 tmp=0 for i in range(n): if 0<=i<r:tmp+=abs(a[i]-a[0]) else:tmp-=abs(a[i]-a[0]) ans=tmp # x=a[i]とし、a[i]から離れている箇所から順に-1にする for i in range(1,n): tmp-=l*(a[i]-a[i-1]) tmp+=(i-l)*(a[i]-a[i-1]) tmp-=(r-i)*(a[i]-a[i-1]) tmp+=(n-r)*(a[i]-a[i-1]) # a[i]から近いk個を+1にし、他は-1 # a[l]とa[r]を比較し、a[l]が近いなら変わらず。a[r]が近いならl+=1,r+=1する。 while r<n and abs(a[l]-a[i])>=abs(a[r]-a[i]): tmp-=2*(a[i]-a[l]) tmp+=2*(a[r]-a[i]) l,r=l+1,r+1 ans=min(ans,tmp) return ans if __name__=='__main__': n,k=map(int,input().split()) a=list(map(int,input().split())) ret1=main1(n,k,a) print(ret1)