結果
問題 |
No.2139 K Consecutive Sushi
|
ユーザー |
![]() |
提出日時 | 2023-10-13 12:25:06 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 261 ms / 2,000 ms |
コード長 | 1,122 bytes |
コンパイル時間 | 497 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 108,416 KB |
最終ジャッジ日時 | 2024-11-27 19:57:15 |
合計ジャッジ時間 | 6,547 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
import sys input = sys.stdin.readline class SegmentTree: def __init__(self, size, f=min, default=10**18): self.size = 2**(size-1).bit_length() self.default = default self.dat = [default]*(self.size*2) self.f = f def update(self, i, x): i += self.size 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 N, K = map(int, input().split()) A = list(map(int, input().split())) T = SegmentTree(N + 1) T.update(0, 0) ans = sum(A) for i in range(N): v = T.query(max(0, i - K + 1), i + 1) T.update(i + 1, v + A[i]) print(ans - T.query(max(0, N - K + 1), N + 1))