結果
問題 | No.2139 K Consecutive Sushi |
ユーザー | ygd. |
提出日時 | 2022-12-05 22:20:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 256 ms / 2,000 ms |
コード長 | 2,166 bytes |
コンパイル時間 | 158 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 111,744 KB |
最終ジャッジ日時 | 2024-05-05 19:23:25 |
合計ジャッジ時間 | 6,353 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 43 ms
53,888 KB |
testcase_01 | AC | 47 ms
53,888 KB |
testcase_02 | AC | 48 ms
54,144 KB |
testcase_03 | AC | 241 ms
106,320 KB |
testcase_04 | AC | 256 ms
105,944 KB |
testcase_05 | AC | 247 ms
105,688 KB |
testcase_06 | AC | 240 ms
106,192 KB |
testcase_07 | AC | 249 ms
105,692 KB |
testcase_08 | AC | 219 ms
106,076 KB |
testcase_09 | AC | 221 ms
106,208 KB |
testcase_10 | AC | 239 ms
106,204 KB |
testcase_11 | AC | 232 ms
105,944 KB |
testcase_12 | AC | 226 ms
106,324 KB |
testcase_13 | AC | 41 ms
53,888 KB |
testcase_14 | AC | 41 ms
53,888 KB |
testcase_15 | AC | 42 ms
53,632 KB |
testcase_16 | AC | 43 ms
53,888 KB |
testcase_17 | AC | 47 ms
54,016 KB |
testcase_18 | AC | 97 ms
76,416 KB |
testcase_19 | AC | 99 ms
76,544 KB |
testcase_20 | AC | 98 ms
76,416 KB |
testcase_21 | AC | 52 ms
58,752 KB |
testcase_22 | AC | 94 ms
76,672 KB |
testcase_23 | AC | 121 ms
78,464 KB |
testcase_24 | AC | 149 ms
85,760 KB |
testcase_25 | AC | 217 ms
103,040 KB |
testcase_26 | AC | 130 ms
78,464 KB |
testcase_27 | AC | 154 ms
84,608 KB |
testcase_28 | AC | 152 ms
84,352 KB |
testcase_29 | AC | 120 ms
79,616 KB |
testcase_30 | AC | 159 ms
88,448 KB |
testcase_31 | AC | 239 ms
102,760 KB |
testcase_32 | AC | 124 ms
80,000 KB |
testcase_33 | AC | 212 ms
111,744 KB |
ソースコード
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()