結果
問題 |
No.2139 K Consecutive Sushi
|
ユーザー |
![]() |
提出日時 | 2022-12-05 22:20:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 240 ms / 2,000 ms |
コード長 | 2,166 bytes |
コンパイル時間 | 325 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 112,256 KB |
最終ジャッジ日時 | 2024-11-27 19:52:52 |
合計ジャッジ時間 | 6,180 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
import sys #input = sys.stdin.readline #input = sys.stdin.buffer.readline #文字列はダメ #sys.setrecursionlimit(1000000) import math import bisect #import itertools #import random #from heapq import heapify, heappop, heappush from collections import defaultdict from collections import deque #import copy #DeepCopy: hoge = [_[:] for _ in hogehoge] #from functools import lru_cache #@lru_cache(maxsize=None) #MOD = pow(10,9) + 7 MOD = 998244353 #dx = [1,0,-1,0] #dy = [0,1,0,-1] #dx8 = [1,1,0,-1,-1,-1,0,1] #dy8 = [0,1,1,1,0,-1,-1,-1] class SegmentTree(object): def __init__(self, A, dot, unit): n = 1 << (len(A) - 1).bit_length() tree = [unit] * (2 * n) for i, v in enumerate(A): tree[i + n] = v for i in range(n - 1, 0, -1): tree[i] = dot(tree[i << 1], tree[i << 1 | 1]) self._n = n self._tree = tree self._dot = dot self._unit = unit def __getitem__(self, i): return self._tree[i + self._n] def update(self, i, v): i += self._n self._tree[i] = v while i != 1: i >>= 1 self._tree[i] = self._dot(self._tree[i << 1], self._tree[i << 1 | 1]) def add(self, i, v): self.update(i, self[i] + v) def sum(self, l, r): #これで[l,r)から取り出す。 l += self._n r += self._n l_val = r_val = self._unit while l < r: if l & 1: l_val = self._dot(l_val, self._tree[l]) l += 1 if r & 1: r -= 1 r_val = self._dot(self._tree[r], r_val) l >>= 1 r >>= 1 return self._dot(l_val, r_val) def main(): N,K = map(int,input().split()); INF = 1<<60 A = list(map(int,input().split())) S = sum(A) # dp[i]: i番目まで見てi番目を食べないとき食べないものの総和 dp = SegmentTree([INF]*(N+1), min, INF) dp.update(0,0) for i in range(1,N+1): dp.update(i,dp.sum(max(i-K,0),i) + A[i-1]) # print(dp[i]) ans = S - dp.sum(N-K+1,N+1) print(ans) if __name__ == '__main__': main()