結果
問題 | No.2139 K Consecutive Sushi |
ユーザー | tassei903 |
提出日時 | 2022-12-03 10:15:44 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 408 ms / 2,000 ms |
コード長 | 2,121 bytes |
コンパイル時間 | 204 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 108,504 KB |
最終ジャッジ日時 | 2024-04-18 18:22:19 |
合計ジャッジ時間 | 8,027 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 38 ms
52,352 KB |
testcase_01 | AC | 40 ms
52,224 KB |
testcase_02 | AC | 40 ms
52,352 KB |
testcase_03 | AC | 394 ms
104,980 KB |
testcase_04 | AC | 394 ms
104,532 KB |
testcase_05 | AC | 392 ms
104,508 KB |
testcase_06 | AC | 396 ms
104,448 KB |
testcase_07 | AC | 408 ms
105,032 KB |
testcase_08 | AC | 328 ms
104,268 KB |
testcase_09 | AC | 326 ms
103,856 KB |
testcase_10 | AC | 345 ms
104,448 KB |
testcase_11 | AC | 329 ms
104,864 KB |
testcase_12 | AC | 317 ms
103,876 KB |
testcase_13 | AC | 39 ms
52,520 KB |
testcase_14 | AC | 38 ms
52,096 KB |
testcase_15 | AC | 41 ms
52,992 KB |
testcase_16 | AC | 39 ms
52,096 KB |
testcase_17 | AC | 38 ms
52,608 KB |
testcase_18 | AC | 94 ms
76,688 KB |
testcase_19 | AC | 97 ms
77,056 KB |
testcase_20 | AC | 94 ms
76,672 KB |
testcase_21 | AC | 45 ms
60,672 KB |
testcase_22 | AC | 112 ms
77,184 KB |
testcase_23 | AC | 158 ms
78,932 KB |
testcase_24 | AC | 222 ms
85,120 KB |
testcase_25 | AC | 307 ms
101,600 KB |
testcase_26 | AC | 163 ms
78,588 KB |
testcase_27 | AC | 211 ms
84,048 KB |
testcase_28 | AC | 198 ms
84,224 KB |
testcase_29 | AC | 171 ms
79,744 KB |
testcase_30 | AC | 233 ms
87,832 KB |
testcase_31 | AC | 363 ms
108,504 KB |
testcase_32 | AC | 181 ms
79,788 KB |
ソースコード
import sys input = lambda :sys.stdin.readline()[:-1] ni = lambda :int(input()) na = lambda :list(map(int,input().split())) yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES") no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO") ####################################################################### inf = 10**18 class SegmentTree: # 初期化処理 # f : SegmentTreeにのせるモノイド # default : fに対する単位元 def __init__(self, size, f=lambda x,y : min(x,y), default=inf): self.size = 2**(size-1).bit_length() # 簡単のため要素数Nを2冪にする self.default = default self.dat = [default]*(self.size*2) # 要素を単位元で初期化 self.f = f def update(self, i, x): i += self.size self.dat[i] = self.f(self.dat[i], x) while i > 0: i >>= 1 self.dat[i] = self.f(self.dat[i*2], self.dat[i*2+1]) def query(self, l, r): l += self.size r += self.size lres, rres = self.default, self.default while l < r: if l & 1: lres = self.f(lres, self.dat[l]) l += 1 if r & 1: r -= 1 rres = self.f(self.dat[r], rres) # モノイドでは可換律は保証されていないので演算の方向に注意 l >>= 1 r >>= 1 res = self.f(lres, rres) return res def query2(self): s = 1 #print(self.size) while s<self.size: #print(s) if self.dat[s*2]>self.dat[s*2+1]: s = s*2 else: s = s*2+1 return s-self.size n, k = na() a = na() dp = SegmentTree(n+1) dp.update(0, 0) for i in range(n): dp.update(i+1, dp.query(max(i - k + 1, 0), i + 1) + a[i]) if i: dp.update(i+1, dp.query(max(i - k + 1, 0), i) + a[i] + a[i-1]) #print(*[dp.query(i, i+1) for i in range(n+1)]) ans = float("inf") for i in range(k): ans = min(ans, dp.query(n-i, n+1-i)) print(sum(a)-ans)